From 593901b2da90f5b2a8c1de897cad45d6c41f99e7 Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Sat, 19 Apr 2025 01:26:21 +0200 Subject: [PATCH] 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 f9b85bf54cc8..ef91e9e6c180 100644 --- a/scripts/convert_wan_to_diffusers.py +++ b/scripts/convert_wan_to_diffusers.py @@ -433,8 +433,9 @@ def get_args(): 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: