Spaces:
Running
on
Zero
Running
on
Zero
Update app_more_lora.py
Browse files- app_more_lora.py +2 -2
app_more_lora.py
CHANGED
@@ -50,9 +50,9 @@ LORA_FILENAME = "FusionX_LoRa/Wan2.1_T2V_14B_FusionX_LoRA.safetensors"
|
|
50 |
|
51 |
vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
|
52 |
wan_path = hf_hub_download(repo_id=SUB_MODEL_ID, filename=SUB_MODEL_FILENAME)
|
53 |
-
transformer = NagWanTransformer3DModel.from_single_file(wan_path, torch_dtype=torch.
|
54 |
pipe = NAGWanPipeline.from_pretrained(
|
55 |
-
MODEL_ID, vae=vae, transformer=transformer, torch_dtype=torch.
|
56 |
)
|
57 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
|
58 |
pipe.to("cuda")
|
|
|
50 |
|
51 |
vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
|
52 |
wan_path = hf_hub_download(repo_id=SUB_MODEL_ID, filename=SUB_MODEL_FILENAME)
|
53 |
+
transformer = NagWanTransformer3DModel.from_single_file(wan_path, torch_dtype=torch.bfloat32)
|
54 |
pipe = NAGWanPipeline.from_pretrained(
|
55 |
+
MODEL_ID, vae=vae, transformer=transformer, torch_dtype=torch.bfloat32
|
56 |
)
|
57 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
|
58 |
pipe.to("cuda")
|