anonymous-aardvark commited on
Commit
d18b63b
1 Parent(s): 4aa3dc9

add opt model

Browse files
Files changed (1) hide show
  1. app.py +7 -4
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
- 'load_fp16' : False,
 
 
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