diff --git a/aiia_vibevoice_nodes.py b/aiia_vibevoice_nodes.py
index 7549dd8..823ed69 100755
--- a/aiia_vibevoice_nodes.py
+++ b/aiia_vibevoice_nodes.py
@@ -146,7 +146,7 @@ class AIIA_VibeVoice_Loader:
         # 5. Load Model
         print(f"[AIIA] Loading VibeVoice model variant: {VibeVoiceClass.__name__}")
         config = VibeVoiceConfig.from_pretrained(load_path)
-        model = VibeVoiceClass.from_pretrained(load_path, config=config, torch_dtype=dtype, trust_remote_code=False).to(device)
+        model = VibeVoiceClass.from_pretrained(load_path, config=config, torch_dtype=dtype, device_map="auto", trust_remote_code=False)
         
         # Load Generation Config
         try:
diff --git a/libs/index-tts/indextts/infer_v2.py b/libs/index-tts/indextts/infer_v2.py
index 77caab0..fc17b70 100644
--- a/libs/index-tts/indextts/infer_v2.py
+++ b/libs/index-tts/indextts/infer_v2.py
@@ -28,7 +28,7 @@ from indextts.s2mel.modules.campplus.DTDNN import CAMPPlus
 from indextts.s2mel.modules.audio import mel_spectrogram
 
 from transformers import AutoTokenizer
-from modelscope import AutoModelForCausalLM
+from transformers import AutoModelForCausalLM
 from huggingface_hub import hf_hub_download
 import safetensors
 from transformers import SeamlessM4TFeatureExtractor