Skip to content

Commit 109e2e7

Browse files
[Bugfix]Fix DMD wan pipeline (hao-ai-lab#659)
1 parent cbc3a6b commit 109e2e7

3 files changed

Lines changed: 2 additions & 2 deletions

File tree

File renamed without changes.

fastvideo/training/wan_distillation_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from fastvideo.logger import init_logger
77
from fastvideo.models.schedulers.scheduling_flow_match_euler_discrete import (
88
FlowMatchEulerDiscreteScheduler)
9-
from fastvideo.pipelines.wan.wan_dmd_pipeline import WanDMDPipeline
9+
from fastvideo.pipelines.basic.wan.wan_dmd_pipeline import WanDMDPipeline
1010
from fastvideo.training.distillation_pipeline import DistillationPipeline
1111
from fastvideo.utils import is_vsa_available
1212

fastvideo/training/wan_i2v_distillation_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from fastvideo.models.schedulers.scheduling_flow_match_euler_discrete import (
1616
FlowMatchEulerDiscreteScheduler)
1717
from fastvideo.pipelines.pipeline_batch_info import ForwardBatch, TrainingBatch
18-
from fastvideo.pipelines.wan.wan_i2v_dmd_pipeline import (
18+
from fastvideo.pipelines.basic.wan.wan_i2v_dmd_pipeline import (
1919
WanImageToVideoDmdPipeline)
2020
from fastvideo.training.distillation_pipeline import DistillationPipeline
2121
from fastvideo.utils import is_vsa_available, shallow_asdict

0 commit comments

Comments
 (0)