Spaces:

tomg-group-umd
/

lm-watermarking

Sleeping

jwkirchenbauer commited on Feb 16, 2023

Commit

8fb55f8

1 Parent(s): 0e6d24f

bigger model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,8 +20,8 @@ args = Namespace()
 arg_dict = {
     'run_gradio': True,
     'demo_public': False,
-    'model_name_or_path': 'facebook/opt-125m',
-    # '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',
+    'model_name_or_path': 'facebook/opt-2.7b',
     'prompt_max_length': None,
     'max_new_tokens': 200,
     'generation_seed': 123,