Skip to content

GitLab

Unverified Commit 5873377a authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

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

update scheuler config in conversion sript
parent 5a2e0f71
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
scripts/convert_wan_to_diffusers.py
View file @ 5873377a
...@@ -433,8 +433,9 @@ if __name__ == "__main__": ...@@ -433,8 +433,9 @@ if __name__ == "__main__":
vae = convert_vae() vae = convert_vae()
text_encoder = UMT5EncoderModel.from_pretrained("google/umt5-xxl") text_encoder = UMT5EncoderModel.from_pretrained("google/umt5-xxl")
tokenizer = AutoTokenizer.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( 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: if "I2V" in args.model_type or "FLF2V" in args.model_type:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment