Spaces:
Sleeping
Sleeping
Commit
·
1a8fdce
1
Parent(s):
7d29596
no share arg
Browse files
app.py
CHANGED
|
@@ -19,14 +19,11 @@ args = Namespace()
|
|
| 19 |
|
| 20 |
arg_dict = {
|
| 21 |
'run_gradio': True,
|
| 22 |
-
|
| 23 |
-
'demo_public': True,
|
| 24 |
'model_name_or_path': 'facebook/opt-125m',
|
| 25 |
# 'model_name_or_path': 'facebook/opt-1.3b',
|
| 26 |
# 'model_name_or_path': 'facebook/opt-2.7b',
|
| 27 |
-
# 'model_name_or_path': 'facebook/opt-6.7b',
|
| 28 |
-
# 'model_name_or_path': 'facebook/opt-13b',
|
| 29 |
-
# 'model_name_or_path': 'facebook/opt-30b',
|
| 30 |
'prompt_max_length': None,
|
| 31 |
'max_new_tokens': 200,
|
| 32 |
'generation_seed': 123,
|
|
@@ -41,7 +38,6 @@ arg_dict = {
|
|
| 41 |
'ignore_repeated_bigrams': False,
|
| 42 |
'detection_z_threshold': 4.0,
|
| 43 |
'select_green_tokens': True,
|
| 44 |
-
# 'skip_model_load': True,
|
| 45 |
'skip_model_load': False,
|
| 46 |
'seed_separately': True,
|
| 47 |
}
|
|
|
|
| 19 |
|
| 20 |
arg_dict = {
|
| 21 |
'run_gradio': True,
|
| 22 |
+
'demo_public': False,
|
|
|
|
| 23 |
'model_name_or_path': 'facebook/opt-125m',
|
| 24 |
# 'model_name_or_path': 'facebook/opt-1.3b',
|
| 25 |
# 'model_name_or_path': 'facebook/opt-2.7b',
|
| 26 |
+
# 'model_name_or_path': 'facebook/opt-6.7b',
|
|
|
|
|
|
|
| 27 |
'prompt_max_length': None,
|
| 28 |
'max_new_tokens': 200,
|
| 29 |
'generation_seed': 123,
|
|
|
|
| 38 |
'ignore_repeated_bigrams': False,
|
| 39 |
'detection_z_threshold': 4.0,
|
| 40 |
'select_green_tokens': True,
|
|
|
|
| 41 |
'skip_model_load': False,
|
| 42 |
'seed_separately': True,
|
| 43 |
}
|