linoyts HF Staff commited on
Commit
ad63290
·
verified ·
1 Parent(s): d7e4b88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -22,6 +22,8 @@ SINGLE_MODAL_VITAL_LAYERS = list(np.array([28, 53, 54, 56, 25]) - 19)
22
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
23
 
24
  torch_dtype=torch.bfloat16)
 
 
25
 
26
  #pipe.enable_lora()
27
  pipe.to("cuda")
 
22
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
23
 
24
  torch_dtype=torch.bfloat16)
25
+ pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
26
+ pipe.fuse_lora(lora_scale=0.125)
27
 
28
  #pipe.enable_lora()
29
  pipe.to("cuda")