LPX55 commited on
Commit
b8b83a5
·
verified ·
1 Parent(s): eb29a27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v
49
  transformer = HunyuanVideoTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.bfloat16)
50
  global pipe
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()
 
49
  transformer = HunyuanVideoTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.bfloat16)
50
  global pipe
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()