anonymous-aardvark
commited on
Commit
•
d18b63b
1
Parent(s):
4aa3dc9
add opt model
Browse files
app.py
CHANGED
@@ -19,8 +19,10 @@ args = Namespace()
|
|
19 |
arg_dict = {
|
20 |
'run_gradio': True,
|
21 |
'demo_public': False,
|
22 |
-
'model_name_or_path': 'bigscience/bloom',
|
23 |
-
'
|
|
|
|
|
24 |
'prompt_max_length': None,
|
25 |
'max_new_tokens': 200,
|
26 |
'generation_seed': 123,
|
@@ -31,11 +33,12 @@ arg_dict = {
|
|
31 |
'seeding_scheme': 'simple_1',
|
32 |
'gamma': 0.5,
|
33 |
'delta': 2.0,
|
34 |
-
'normalizers': '',
|
35 |
'ignore_repeated_bigrams': False,
|
36 |
'detection_z_threshold': 4.0,
|
37 |
'select_green_tokens': True,
|
38 |
-
'skip_model_load': True,
|
|
|
39 |
'seed_separately': True,
|
40 |
}
|
41 |
|
|
|
19 |
arg_dict = {
|
20 |
'run_gradio': True,
|
21 |
'demo_public': False,
|
22 |
+
# 'model_name_or_path': 'bigscience/bloom',
|
23 |
+
'model_name_or_path': 'facebook/opt-6.7b',
|
24 |
+
# 'load_fp16' : False,
|
25 |
+
'load_fp16' : True,
|
26 |
'prompt_max_length': None,
|
27 |
'max_new_tokens': 200,
|
28 |
'generation_seed': 123,
|
|
|
33 |
'seeding_scheme': 'simple_1',
|
34 |
'gamma': 0.5,
|
35 |
'delta': 2.0,
|
36 |
+
'normalizers': '',
|
37 |
'ignore_repeated_bigrams': False,
|
38 |
'detection_z_threshold': 4.0,
|
39 |
'select_green_tokens': True,
|
40 |
+
# 'skip_model_load': True,
|
41 |
+
'skip_model_load': False,
|
42 |
'seed_separately': True,
|
43 |
}
|
44 |
|