Crystalcareai commited on
Commit
a5c6c16
·
verified ·
1 Parent(s): e35584d

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +1 -1
generate.py CHANGED
@@ -137,7 +137,7 @@ def generate(
137
  forced_eos_token_id=None,
138
  remove_invalid_values=None,
139
  synced_gpus=None,
140
- n_ahead=4,
141
  n_ahead_talk=4,
142
  merged_talk_heads=True,
143
  merged_lm_and_talk_heads=False,
 
137
  forced_eos_token_id=None,
138
  remove_invalid_values=None,
139
  synced_gpus=None,
140
+ n_ahead=8,
141
  n_ahead_talk=4,
142
  merged_talk_heads=True,
143
  merged_lm_and_talk_heads=False,