cyqfwang commited on
Commit
9c73b03
·
verified ·
1 Parent(s): 05334cb

Update generation_utils.py

Browse files

The `validate` method in `DreamGenerationConfig` misses a argument in base class `‎GenerationConfig`.
cf. https://huggingface.co/docs/transformers/en/main_classes/text_generation#transformers.GenerationConfig.validate

Files changed (1) hide show
  1. generation_utils.py +1 -1
generation_utils.py CHANGED
@@ -143,7 +143,7 @@ class DreamGenerationConfig(GenerationConfig):
143
  # Validate the values of the attributes
144
  self.validate(is_init=True)
145
 
146
- def validate(self, is_init=False):
147
  pass
148
 
149
  class DreamGenerationMixin:
 
143
  # Validate the values of the attributes
144
  self.validate(is_init=True)
145
 
146
+ def validate(self, is_init=False, strict=False):
147
  pass
148
 
149
  class DreamGenerationMixin: