LPX55 commited on
Commit
f3dde8f
·
verified ·
1 Parent(s): 1ed6589

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ global pipe
51
  pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.bfloat16)
52
  pipe.to("cuda")
53
  # Enable memory savings
54
- # pipe.vae.enable_slicing()
55
- # pipe.vae.enable_tiling()
56
  # pipe.enable_model_cpu_offload()
57
 
58
  with torch.no_grad(): # enable image inputs
 
51
  pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.bfloat16)
52
  pipe.to("cuda")
53
  # Enable memory savings
54
+ pipe.vae.enable_slicing()
55
+ pipe.vae.enable_tiling()
56
  # pipe.enable_model_cpu_offload()
57
 
58
  with torch.no_grad(): # enable image inputs