minor fix

SWivid · Oct 22, 2024 · 752f6f5 · 752f6f5
1 parent cd3c4af
commit 752f6f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/model/utils_infer.py b/model/utils_infer.py
@@ -154,7 +154,8 @@ def load_model(model_cls, model_cfg, ckpt_path, vocab_file="", ode_method=ode_me
 
 
 def preprocess_ref_audio_text(ref_audio_orig, ref_text, show_info=print, device=None):
-    device = get_device(device)
+    if device is None:
+        device = get_device()
 
     show_info("Converting audio...")
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f: