Spaces:

tomg-group-umd
/

lm-watermarking

Sleeping

jwkirchenbauer commited on Feb 17, 2023

Commit

a982455

1 Parent(s): 1a8fdce

up model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,9 +20,9 @@ args = Namespace()
 arg_dict = {
     'run_gradio': True,
     'demo_public': False,
-    'model_name_or_path': 'facebook/opt-125m',
     # 'model_name_or_path': 'facebook/opt-1.3b',
-    # 'model_name_or_path': 'facebook/opt-2.7b',
     # 'model_name_or_path': 'facebook/opt-6.7b',
     'prompt_max_length': None,
     'max_new_tokens': 200,

 arg_dict = {
     'run_gradio': True,
     'demo_public': False,
+    # 'model_name_or_path': 'facebook/opt-125m',
     # 'model_name_or_path': 'facebook/opt-1.3b',
+    'model_name_or_path': 'facebook/opt-2.7b',
     # 'model_name_or_path': 'facebook/opt-6.7b',
     'prompt_max_length': None,
     'max_new_tokens': 200,