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 |
|