LPX55 commited on
Commit
d1fc296
·
verified ·
1 Parent(s): 5a8cf56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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