farzadab commited on
Commit
5190dda
·
verified ·
1 Parent(s): bb86f58

Upload ultravox_model.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. ultravox_model.py +2 -0
ultravox_model.py CHANGED
@@ -79,6 +79,8 @@ class UltravoxModel(transformers.LlamaPreTrainedModel, GenerationMixin):
79
  if "torch_dtype" in kwargs:
80
  self.config.torch_dtype = kwargs.pop("torch_dtype")
81
 
 
 
82
  if (
83
  self.config.text_model_id is not None
84
  and self.language_model.device.type == "meta"
 
79
  if "torch_dtype" in kwargs:
80
  self.config.torch_dtype = kwargs.pop("torch_dtype")
81
 
82
+ kwargs.pop("config", None)
83
+
84
  if (
85
  self.config.text_model_id is not None
86
  and self.language_model.device.type == "meta"