From 135c98c086cddd1d50ea81c4de2d5d732d879267 Mon Sep 17 00:00:00 2001 From: zhaoyun007 <35762050+zhaoyun0071@users.noreply.github.com> Date: Sun, 1 Feb 2026 13:16:27 +0800 Subject: [PATCH 1/2] Import WanTransformer3DModel from diffusers.models --- telestylevideo_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 From 12d8a9bd884b6b72afa492dda7d4456c93ad7bc0 Mon Sep 17 00:00:00 2001 From: zhaoyun007 <35762050+zhaoyun0071@users.noreply.github.com> Date: Sun, 1 Feb 2026 13:19:26 +0800 Subject: [PATCH 2/2] Update transformer config path to new structure --- telestylevideo_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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']