From 733c14a2cb460e29b80d6c94d4a549a9dbd52d35 Mon Sep 17 00:00:00 2001 From: JerryZhou54 Date: Sun, 16 Nov 2025 04:53:57 +0000 Subject: [PATCH] Change to fp32 inference --- .../inference/basic/basic_self_forcing_causal_wan2_2_i2v.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/inference/basic/basic_self_forcing_causal_wan2_2_i2v.py b/examples/inference/basic/basic_self_forcing_causal_wan2_2_i2v.py index 4806de7c0..c8fcc6407 100644 --- a/examples/inference/basic/basic_self_forcing_causal_wan2_2_i2v.py +++ b/examples/inference/basic/basic_self_forcing_causal_wan2_2_i2v.py @@ -16,6 +16,7 @@ def main(): num_gpus=1, use_fsdp_inference=True, dit_cpu_offload=True, # DiT need to be offloaded for MoE + dit_precision="fp32", vae_cpu_offload=False, text_encoder_cpu_offload=True, dmd_denoising_steps=[1000, 850, 700, 550, 350, 275, 200, 125], @@ -30,7 +31,7 @@ def main(): sampling_param.num_frames = 73 sampling_param.width = 832 sampling_param.height = 480 - sampling_param.seed = 42 + sampling_param.seed = 1000 with open("prompts/mixkit_i2v.jsonl", "r") as f: prompt_image_pairs = json.load(f)