AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Wan2.1-FLF2V] update conversion script (#11365)

Browse Source 
update scheuler config in conversion sript
This commit is contained in:
YiYi Xu
2025-04-18 14:08:44 -10:00
committed by GitHub
parent 5a2e0f715c
commit 5873377a66
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/convert_wan_to_diffusers.py
View File

@@ -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: