NULLNode commited on
Commit
719d4da
1 Parent(s): 893a1df

Update app.py

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