Update modeling_diva.py
Browse files- modeling_diva.py +1 -1
modeling_diva.py
CHANGED
@@ -143,7 +143,7 @@ class DiVAModel(PreTrainedModel):
|
|
143 |
return cls(
|
144 |
via_path,
|
145 |
config_dict,
|
146 |
-
kwargs["device_map"] if "device_map" in kwargs else
|
147 |
(
|
148 |
kwargs["speech_encoder_device"]
|
149 |
if "speech_encoder_device" in kwargs
|
|
|
143 |
return cls(
|
144 |
via_path,
|
145 |
config_dict,
|
146 |
+
kwargs["device_map"] if "device_map" in kwargs else "auto",
|
147 |
(
|
148 |
kwargs["speech_encoder_device"]
|
149 |
if "speech_encoder_device" in kwargs
|