linoyts HF Staff commited on
Commit
31944ca
·
verified ·
1 Parent(s): 7176730

Update app.py

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