Feature Extraction
Transformers
Safetensors
diva
custom_code
WillHeld commited on
Commit
2a901db
1 Parent(s): 323fae5

Update modeling_diva.py

Browse files
Files changed (1) hide show
  1. 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 None,
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