Spaces:

markury
/

Wan-2.1-T2V-1.3B-LoRA

Running on Zero

markury commited on Mar 19

Commit

1ead6f2

1 Parent(s): 425d66f

load transformer in fp16

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ def generate_video(
     # Load model
     vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
-    pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
     # Set scheduler
     if scheduler_type == "UniPCMultistepScheduler":

     # Load model
     vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
+    pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.float16)
     # Set scheduler
     if scheduler_type == "UniPCMultistepScheduler":