Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,8 @@ arg_dict = {
|
|
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 |
# 'model_name_or_path': 'facebook/opt-13b',
|
28 |
# 'load_fp16' : True,
|
29 |
'load_fp16' : False,
|
@@ -41,7 +42,7 @@ arg_dict = {
|
|
41 |
'ignore_repeated_bigrams': False,
|
42 |
'detection_z_threshold': 4.0,
|
43 |
'select_green_tokens': True,
|
44 |
-
'skip_model_load':
|
45 |
'seed_separately': True,
|
46 |
}
|
47 |
|
|
|
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 |
+
'model_name_or_path': 'facebook/opt-125m',
|
28 |
# 'model_name_or_path': 'facebook/opt-13b',
|
29 |
# 'load_fp16' : True,
|
30 |
'load_fp16' : False,
|
|
|
42 |
'ignore_repeated_bigrams': False,
|
43 |
'detection_z_threshold': 4.0,
|
44 |
'select_green_tokens': True,
|
45 |
+
'skip_model_load': False,
|
46 |
'seed_separately': True,
|
47 |
}
|
48 |
|