Update modeling_dream.py
Browse files- 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 |
-
|
| 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 |
|