ovi054 commited on
Commit
8d5e30a
·
verified ·
1 Parent(s): 95b6159

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
27
  model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
28
  # model_id_21 = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
29
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
30
- vae.enable_tiling()
31
- vae.enable_slicing()
32
  transformer = WanTransformer3DModel.from_pretrained(model_id, subfolder="transformer_2", torch_dtype=torch.bfloat16)
33
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16)
34
  flow_shift = 1.0 #5.0 1.0 for image, 5.0 for 720P, 3.0 for 480P
 
27
  model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
28
  # model_id_21 = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
29
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
30
+ # vae.enable_tiling()
31
+ # vae.enable_slicing()
32
  transformer = WanTransformer3DModel.from_pretrained(model_id, subfolder="transformer_2", torch_dtype=torch.bfloat16)
33
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16)
34
  flow_shift = 1.0 #5.0 1.0 for image, 5.0 for 720P, 3.0 for 480P