smoorsmith commited on
Commit
b86108d
·
verified ·
1 Parent(s): 2602c18

Update modeling_dream.py

Browse files
Files changed (1) hide show
  1. modeling_dream.py +1 -1
modeling_dream.py CHANGED
@@ -754,7 +754,7 @@ class DreamModel(DreamGenerationMixin, DreamPreTrainedModel):
754
  input_ids = input_ids[:, -1:]
755
  model_inputs = {"input_ids": input_ids, "past_key_values": past_key_values}
756
 
757
- model_inputs.update(kwargs)
758
  model_inputs["use_cache"] = kwargs.pop("use_cache", self.config.use_cache)
759
  return model_inputs
760
 
 
754
  input_ids = input_ids[:, -1:]
755
  model_inputs = {"input_ids": input_ids, "past_key_values": past_key_values}
756
 
757
+ model_inputs.update(kwargs)
758
  model_inputs["use_cache"] = kwargs.pop("use_cache", self.config.use_cache)
759
  return model_inputs
760