Commit 90e312f3 authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

Update wan_runner.py

parent de3ada69
......@@ -464,7 +464,7 @@ class Wan22DenseRunner(WanRunner):
"cpu_offload": vae_offload,
"offload_cache": self.config.get("vae_offload_cache", False),
}
if self.config.task != ["i2v", "flf2v"]:
if self.config.task not in ["i2v", "flf2v"]:
return None
else:
return Wan2_2_VAE(**vae_config)
......
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