LPX55 commited on
Commit
6180d48
·
verified ·
1 Parent(s): e0bc09a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -50,9 +50,9 @@ transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(
50
 
51
  pipeline = HunyuanVideoPipeline.from_pretrained(
52
  "hunyuanvideo-community/HunyuanVideo",
53
- transformer=transformer_8bit,
54
  torch_dtype=torch.float16,
55
- device_map="balanced",
56
  )
57
  model_id = "hunyuanvideo-community/HunyuanVideo"
58
  # lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v.sft")
 
50
 
51
  pipeline = HunyuanVideoPipeline.from_pretrained(
52
  "hunyuanvideo-community/HunyuanVideo",
53
+ # transformer=transformer_8bit,
54
  torch_dtype=torch.float16,
55
+ # device_map="balanced",
56
  )
57
  model_id = "hunyuanvideo-community/HunyuanVideo"
58
  # lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v.sft")