Crystalcareai commited on
Commit
4e0f647
·
verified ·
1 Parent(s): e1037c9

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +6 -6
generate.py CHANGED
@@ -162,12 +162,12 @@ def generate(
162
  self.rm_initialized = True
163
  self.original_mode = False
164
 
165
- # Validate stopping criteria
166
- stopping_criteria = self._get_stopping_criteria(generation_config=self.config, stopping_criteria=StoppingCriteriaList())
167
- if stopping_criteria is None:
168
- stopping_criteria = StoppingCriteriaList()
169
- if max_length is not None:
170
- stopping_criteria = validate_stopping_criteria(max_length, stopping_criteria=stopping_criteria)
171
  # Generate using the custom generate function
172
  generated_token_ids = custom_generate(
173
  self,
 
162
  self.rm_initialized = True
163
  self.original_mode = False
164
 
165
+ # # Validate stopping criteria
166
+ # stopping_criteria = self._get_stopping_criteria(generation_config=self.config, stopping_criteria=StoppingCriteriaList())
167
+ # if stopping_criteria is None:
168
+ # stopping_criteria = StoppingCriteriaList()
169
+ # if max_length is not None:
170
+ # stopping_criteria = validate_stopping_criteria(max_length, stopping_criteria=stopping_criteria)
171
  # Generate using the custom generate function
172
  generated_token_ids = custom_generate(
173
  self,