kontext-relight

Running on Zero

linoyts HF Staff commited on 4 days ago

Commit

31944ca

verified ·

1 Parent(s): 7176730

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from diffusers.utils import load_image
 from huggingface_hub import hf_hub_download
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
-pipe.load_lora_weights("kontext-community/relighting-kontext-dev-lora-v3", weight_name="relighting-kontext-dev-lora-v3.safetensors", adapter_name="lora")
 pipe.set_adapters(["lora"], adapter_weights=[1.0])
 MAX_SEED = np.iinfo(np.int32).max

 from huggingface_hub import hf_hub_download
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
+pipe.load_lora_weights("kontext-community/relighting-kontext-dev-lora-v2", weight_name="relighting-kontext-dev-lora.safetensors", adapter_name="lora")
 pipe.set_adapters(["lora"], adapter_weights=[1.0])
 MAX_SEED = np.iinfo(np.int32).max