multimodalart HF Staff commited on
Commit
e7b9df9
·
verified ·
1 Parent(s): 98b03ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -155,9 +155,10 @@ pipe = SkyReelsA1ImagePoseToVideoPipeline.from_pretrained(
155
  )
156
  pipe.to("cuda")
157
  pipe.transformer = torch.compile(pipe.transformer)
 
 
158
  pipe.vae = torch.compile(pipe.vae)
159
  # pipe.enable_model_cpu_offload()
160
- # pipe.vae.enable_tiling()
161
 
162
  print("Models loaded successfully!")
163
 
 
155
  )
156
  pipe.to("cuda")
157
  pipe.transformer = torch.compile(pipe.transformer)
158
+ pipe.vae.enable_tiling()
159
+
160
  pipe.vae = torch.compile(pipe.vae)
161
  # pipe.enable_model_cpu_offload()
 
162
 
163
  print("Models loaded successfully!")
164