diff --git a/telestylevideo_inference.py b/telestylevideo_inference.py
index f154213..b8091c2 100755
--- a/telestylevideo_inference.py
+++ b/telestylevideo_inference.py
@@ -55,7 +55,7 @@ def __init__(self, config: Dict):
         self.W = config['width']
         self.num_inference_steps = config['num_inference_steps']
         self.vae_path = os.path.join(config['ckpt_t2v_path'], "vae")
-        self.transformer_config_path = os.path.join(config['ckpt_t2v_path'], "transformer_config.json")
+        self.transformer_config_path = os.path.join(config['ckpt_t2v_path'], "transformer", "config.json")
         self.scheduler_path = os.path.join(config['ckpt_t2v_path'], "scheduler")
         self.ckpt_path = config['ckpt_dit_path']
         self.output_path = config['output_path']
diff --git a/telestylevideo_pipeline.py b/telestylevideo_pipeline.py
index d6c931a..bb6a971 100755
--- a/telestylevideo_pipeline.py
+++ b/telestylevideo_pipeline.py
@@ -23,7 +23,7 @@
 from diffusers.callbacks import MultiPipelineCallbacks, PipelineCallback
 from diffusers.loaders import WanLoraLoaderMixin
 from diffusers.models import AutoencoderKLWan
-from transformer_semi_dit_2_patch_embedders import WanTransformer3DModel
+from diffusers.models import WanTransformer3DModel
 from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
 from diffusers.utils import is_torch_xla_available, logging, replace_example_docstring
 from diffusers.utils.torch_utils import randn_tensor