linoyts HF Staff commited on
Commit
fbe8913
·
verified ·
1 Parent(s): 4aa3956

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -12,6 +12,8 @@ from diffusers.utils import load_image
12
  from huggingface_hub import hf_hub_download
13
 
14
  pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 
 
15
 
16
  MAX_SEED = np.iinfo(np.int32).max
17
 
 
12
  from huggingface_hub import hf_hub_download
13
 
14
  pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
15
+ pipe.load_lora_weights("linoyts/relighting-kontext-dev-lora", weight_name="relighting-kontext-dev-lora.safetensors", adapter_name="lora")
16
+ pipe.set_adapters(["lora"], adapter_weights=[1.0])
17
 
18
  MAX_SEED = np.iinfo(np.int32).max
19