linoyts HF Staff commited on
Commit
36ec274
·
verified ·
1 Parent(s): 87c8ea9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +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
- pipe.load_lora_weights("kontext-community/relighting-kontext-dev-lora-v2", 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
 
 
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("kontext-community/relighting-kontext-dev-lora-v3", weight_name="relighting-kontext-dev-lora-v3.safetensors", adapter_name="lora")
16
+ pipe.set_adapters(["lora"], adapter_weights=[0.75])
17
 
18
  MAX_SEED = np.iinfo(np.int32).max
19