From 5873377a660dac60a6bd86ef9b4fdfc385305977 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Fri, 18 Apr 2025 14:08:44 -1000 Subject: [PATCH] [Wan2.1-FLF2V] update conversion script (#11365) update scheuler config in conversion sript --- scripts/convert_wan_to_diffusers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/convert_wan_to_diffusers.py b/scripts/convert_wan_to_diffusers.py index f9b85bf54c..ef91e9e6c1 100644 --- a/scripts/convert_wan_to_diffusers.py +++ b/scripts/convert_wan_to_diffusers.py @@ -433,8 +433,9 @@ if __name__ == "__main__": vae = convert_vae() text_encoder = UMT5EncoderModel.from_pretrained("google/umt5-xxl") tokenizer = AutoTokenizer.from_pretrained("google/umt5-xxl") + flow_shift = 16.0 if "FLF2V" in args.model_type else 3.0 scheduler = UniPCMultistepScheduler( - prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=3.0 + prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=flow_shift ) if "I2V" in args.model_type or "FLF2V" in args.model_type: