Skip to content

Commit 5873377

Browse files
authored
[Wan2.1-FLF2V] update conversion script (#11365)
update scheuler config in conversion sript
1 parent 5a2e0f7 commit 5873377

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

scripts/convert_wan_to_diffusers.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -433,8 +433,9 @@ def get_args():
433433
vae = convert_vae()
434434
text_encoder = UMT5EncoderModel.from_pretrained("google/umt5-xxl")
435435
tokenizer = AutoTokenizer.from_pretrained("google/umt5-xxl")
436+
flow_shift = 16.0 if "FLF2V" in args.model_type else 3.0
436437
scheduler = UniPCMultistepScheduler(
437-
prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=3.0
438+
prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=flow_shift
438439
)
439440

440441
if "I2V" in args.model_type or "FLF2V" in args.model_type:

0 commit comments

Comments
 (0)