diff --git a/vibevoice_core/modular/modeling_vibevoice.py b/vibevoice_core/modular/modeling_vibevoice.py
index 54c7c94..b2c27e8 100644
--- a/vibevoice_core/modular/modeling_vibevoice.py
+++ b/vibevoice_core/modular/modeling_vibevoice.py
@@ -20,8 +20,8 @@ from transformers.modeling_flash_attention_utils import FlashAttentionKwargs
 from transformers.utils import logging
 
 
-from .modular_vibevoice_tokenizer import VibeVoiceTokenizerStreamingCache, VibeVoiceAcousticTokenizerModel, VibeVoiceSemanticTokenizerModel
-from .modular_vibevoice_diffusion_head import VibeVoiceDiffusionHead
+from modular_vibevoice_tokenizer import VibeVoiceTokenizerStreamingCache, VibeVoiceAcousticTokenizerModel, VibeVoiceSemanticTokenizerModel
+from modular_vibevoice_diffusion_head import VibeVoiceDiffusionHead
 
 # Import schedule module with robust path handling to avoid conflicts with PyPI 'schedule' package
 import sys
diff --git a/vibevoice_core/modular/modeling_vibevoice_inference.py b/vibevoice_core/modular/modeling_vibevoice_inference.py
index 286deb5..f510d0a 100644
--- a/vibevoice_core/modular/modeling_vibevoice_inference.py
+++ b/vibevoice_core/modular/modeling_vibevoice_inference.py
@@ -18,8 +18,8 @@ from transformers.utils import logging
 
 
 # from .modular_vibevoice_tokenizer import VibeVoiceTokenizerStreamingCache, VibeVoiceAcousticTokenizerModel, VibeVoiceSemanticTokenizerModel
-from .modular_vibevoice_tokenizer import VibeVoiceTokenizerStreamingCache, VibeVoiceTokenizerEncoderOutput
-from .modular_vibevoice_diffusion_head import VibeVoiceDiffusionHead
+from modular_vibevoice_tokenizer import VibeVoiceTokenizerStreamingCache, VibeVoiceTokenizerEncoderOutput
+from modular_vibevoice_diffusion_head import VibeVoiceDiffusionHead
 
 # Import schedule module with robust path handling to avoid conflicts with PyPI 'schedule' package
 import sys