Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,6 +37,7 @@ video_transforms = transforms.Compose(
|
|
| 37 |
model_id = "hunyuanvideo-community/HunyuanVideo"
|
| 38 |
lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v.sft") # Replace with the actual LORA path
|
| 39 |
transformer = HunyuanVideoTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.bfloat16)
|
|
|
|
| 40 |
pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.bfloat16)
|
| 41 |
|
| 42 |
# Enable memory savings
|
|
|
|
| 37 |
model_id = "hunyuanvideo-community/HunyuanVideo"
|
| 38 |
lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v.sft") # Replace with the actual LORA path
|
| 39 |
transformer = HunyuanVideoTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.bfloat16)
|
| 40 |
+
global pipe
|
| 41 |
pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.bfloat16)
|
| 42 |
|
| 43 |
# Enable memory savings
|