rahul7star commited on
Commit
c3f8fd8
·
verified ·
1 Parent(s): a491fdc

Update app_more_lora.py

Browse files
Files changed (1) hide show
  1. 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.bfloat16)
54
  pipe = NAGWanPipeline.from_pretrained(
55
- MODEL_ID, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16
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")