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
|