Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
55 |
-
|
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
|