Draw2Photo

Runtime error

ovi054 commited on Sep 25

Commit

fde9193

verified ·

1 Parent(s): 1016668

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ from diffusers.utils import load_image
 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
-pipe.load_lora_weights("ovi054/Draw2Photo")
 pipe.fuse_lora()
 # optimize_pipeline_(pipe, image=Image.new("RGB", (512, 512)), prompt='prompt')

 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
+pipe.load_lora_weights("ovi054/Draw2Photo-Kontext-LoRA")
 pipe.fuse_lora()
 # optimize_pipeline_(pipe, image=Image.new("RGB", (512, 512)), prompt='prompt')