Skip to content

Commit 9d7d1ca

Browse files
committed
[fix]: restore Lucy review fixes
1 parent ba074df commit 9d7d1ca

3 files changed

Lines changed: 12 additions & 0 deletions

File tree

fastvideo/configs/pipelines/wan.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ class Wan2_2_TI2V_5B_Config(WanT2V480PConfig):
122122
expand_timesteps: bool = True
123123

124124
def __post_init__(self) -> None:
125+
assert not (self.ti2v_task and self.lucy_edit_task)
125126
self.vae_config.load_encoder = True
126127
self.vae_config.load_decoder = True
127128
self.dit_config.expand_timesteps = self.expand_timesteps
@@ -173,6 +174,7 @@ class LucyEditDevConfig(Wan2_2_TI2V_5B_Config):
173174
lucy_edit_task: bool = True
174175

175176
def __post_init__(self) -> None:
177+
assert not (self.ti2v_task and self.lucy_edit_task)
176178
# Lucy uses Wan2.2's enhanced 48-channel VAE latents. Denoising
177179
# concatenates noise + video latents, matching the 96-channel
178180
# transformer input declared above.

fastvideo/pipelines/basic/wan/presets.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -273,6 +273,7 @@
273273
version=1,
274274
model_family="wan",
275275
description="Lucy Edit Dev 5B video editing",
276+
workload_type="t2v",
276277
stage_schemas=(_DENOISE_STAGE, ),
277278
defaults={
278279
"height": 480,

fastvideo/tests/api/test_lucy_edit.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ def test_lucy_edit_registry_and_preset() -> None:
2828
assert get_default_preset("decart-ai/Lucy-Edit-Dev") == "lucy_edit_dev"
2929
assert get_default_preset("decart-ai/Lucy-Edit-1.1-Dev") == "lucy_edit_dev"
3030
assert get_pipeline_config_cls_from_name("decart-ai/Lucy-Edit-Dev") is LucyEditDevConfig
31+
assert get_pipeline_config_cls_from_name("decart-ai/Lucy-Edit-1.1-Dev") is LucyEditDevConfig
3132

3233
sampling_param = SamplingParam.from_pretrained("decart-ai/Lucy-Edit-Dev")
3334
assert sampling_param.height == 480
@@ -37,6 +38,14 @@ def test_lucy_edit_registry_and_preset() -> None:
3738
assert sampling_param.guidance_scale == 5.0
3839
assert sampling_param.negative_prompt == ""
3940

41+
sampling_param_1_1 = SamplingParam.from_pretrained("decart-ai/Lucy-Edit-1.1-Dev")
42+
assert sampling_param_1_1.height == 480
43+
assert sampling_param_1_1.width == 832
44+
assert sampling_param_1_1.num_frames == 81
45+
assert sampling_param_1_1.fps == 24
46+
assert sampling_param_1_1.guidance_scale == 5.0
47+
assert sampling_param_1_1.negative_prompt == ""
48+
4049
# FastVideo has no V2V workload enum today; model_index dispatches Lucy by pipeline class name.
4150
registry = get_pipeline_registry(PipelineType.BASIC)
4251
assert registry.resolve_pipeline_cls("LucyEditPipeline", PipelineType.BASIC, WorkloadType.T2V) is LucyEditPipeline

0 commit comments

Comments
 (0)