Crystalcareai commited on
Commit
8b674a4
·
verified ·
1 Parent(s): 9a29d47

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +1 -2
generate.py CHANGED
@@ -164,9 +164,8 @@ def generate(
164
 
165
  # Validate stopping criteria
166
  stopping_criteria = validate_stopping_criteria(
167
- max_length, max_new_tokens=None, min_length=min_length, stopping_criteria=self.stopping_criteria
168
  )
169
-
170
  # Generate using the custom generate function
171
  generated_token_ids = custom_generate(
172
  self,
 
164
 
165
  # Validate stopping criteria
166
  stopping_criteria = validate_stopping_criteria(
167
+ max_length, max_new_tokens=None, min_length=min_length, stopping_criteria=self._get_stopping_criteria()
168
  )
 
169
  # Generate using the custom generate function
170
  generated_token_ids = custom_generate(
171
  self,