File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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.
Original file line number Diff line number Diff line change 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 ,
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments