Spaces:

tomg-group-umd
/

lm-watermarking

Sleeping

jwkirchenbauer commited on Feb 17, 2023

Commit

cb2cffc

•

1 Parent(s): 99fe45a

downsize model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,8 +22,8 @@ arg_dict = {
     '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,
     'generation_seed': 123,

     '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,
     'generation_seed': 123,