image-i1-dev2

Runtime error

ovi054 commited on Aug 10

Commit

43ce188

verified ·

1 Parent(s): 62f540c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import spaces
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # --- MODEL SETUP ---
-model_id = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 flow_shift = 1.0

 device = "cuda" if torch.cuda.is_available() else "cpu"
 # --- MODEL SETUP ---
+model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 flow_shift = 1.0