multimodalart HF staff commited on
Commit
85f2c6e
1 Parent(s): 29dc892

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -19,6 +19,10 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
19
  transformer=transformer,
20
  torch_dtype=torch.bfloat16
21
  ).to("cuda")
 
 
 
 
22
  pipe.transformer.to(torch.bfloat16)
23
  pipe.controlnet.to(torch.bfloat16)
24
 
 
19
  transformer=transformer,
20
  torch_dtype=torch.bfloat16
21
  ).to("cuda")
22
+ repo_name = "ByteDance/Hyper-SD"
23
+ ckpt_name = "Hyper-FLUX.1-dev-8steps-lora.safetensors"
24
+ pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
25
+ pipe.fuse_lora(lora_scale=0.125)
26
  pipe.transformer.to(torch.bfloat16)
27
  pipe.controlnet.to(torch.bfloat16)
28