lm-watermarking

Runtime error

jwkirchenbauer commited on Feb 16, 2023

Commit

02d9c9c

1 Parent(s): a485f37

drop model size again

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ args = Namespace()
 arg_dict = {
     'run_gradio': True,
     'demo_public': False,
-    'model_name_or_path': 'facebook/opt-2.7b',
     'prompt_max_length': None,
     'max_new_tokens': 200,
     'generation_seed': 123,

 arg_dict = {
     'run_gradio': True,
     'demo_public': False,
+    'model_name_or_path': 'facebook/opt-125m',
     'prompt_max_length': None,
     'max_new_tokens': 200,
     'generation_seed': 123,