Update scripts/convert_wan_to_diffusers.py

a-r-r-o-w · web-flow · commit 36297943c50a · 2025-03-02T16:31:16.000+05:30
diff --git a/scripts/convert_wan_to_diffusers.py b/scripts/convert_wan_to_diffusers.py
@@ -394,7 +394,7 @@ def get_args():
     text_encoder = UMT5EncoderModel.from_pretrained("google/umt5-xxl")
     tokenizer = AutoTokenizer.from_pretrained("google/umt5-xxl")
     scheduler = UniPCMultistepScheduler(
-        prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=1.0
+        prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=3.0
     )
 
     if "I2V" in args.model_type:

Original file line number	Diff line number	Diff line change
`@@ -394,7 +394,7 @@ def get_args():`
`394`	`394`	`text_encoder = UMT5EncoderModel.from_pretrained("google/umt5-xxl")`
`395`	`395`	`tokenizer = AutoTokenizer.from_pretrained("google/umt5-xxl")`
`396`	`396`	`scheduler = UniPCMultistepScheduler(`
`397`		`- prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=1.0`
	`397`	`+ prediction_type="flow_prediction", use_flow_sigmas=True, num_train_timesteps=1000, flow_shift=3.0`
`398`	`398`	`)`
`399`	`399`
`400`	`400`	`if "I2V" in args.model_type:`