Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
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 |
|