Spaces:
Runtime error
Runtime error
opt
Browse files
app.py
CHANGED
|
@@ -48,8 +48,9 @@ lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v
|
|
| 48 |
transformer = HunyuanVideoTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.bfloat16)
|
| 49 |
global pipe
|
| 50 |
pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.bfloat16)
|
| 51 |
-
|
| 52 |
# Enable memory savings
|
|
|
|
| 53 |
pipe.vae.enable_tiling()
|
| 54 |
pipe.enable_model_cpu_offload()
|
| 55 |
|
|
|
|
| 48 |
transformer = HunyuanVideoTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.bfloat16)
|
| 49 |
global pipe
|
| 50 |
pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.bfloat16)
|
| 51 |
+
pipe.to("cuda")
|
| 52 |
# Enable memory savings
|
| 53 |
+
pipe.vae.enable_slicing()
|
| 54 |
pipe.vae.enable_tiling()
|
| 55 |
pipe.enable_model_cpu_offload()
|
| 56 |
|