Crystalcareai commited on
Commit
7073ab6
·
verified ·
1 Parent(s): dc34aea

Update modeling_quiet.py

Browse files
Files changed (1) hide show
  1. modeling_quiet.py +1 -1
modeling_quiet.py CHANGED
@@ -1424,7 +1424,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
1424
 
1425
  def generate(self, input_ids, attention_mask=None, max_length=None, temperature=1.0, **kwargs):
1426
  from .generate import generate
1427
- return generate(self, input_ids, attention_mask=attention_mask, max_length=max_length, temperature=temperature, **kwargs)
1428
 
1429
  @add_start_docstrings_to_model_forward(QUIET_INPUTS_DOCSTRING)
1430
  @replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
 
1424
 
1425
  def generate(self, input_ids, attention_mask=None, max_length=None, temperature=1.0, **kwargs):
1426
  from .generate import generate
1427
+ return generate(self, input_ids, attention_mask, max_length, temperature, **kwargs)
1428
 
1429
  @add_start_docstrings_to_model_forward(QUIET_INPUTS_DOCSTRING)
1430
  @replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)