Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -238,6 +238,8 @@ def generate_mask(inp_image, label, threshold):
|
|
238 |
|
239 |
#Setting up Flux (Schnell) Inpainting
|
240 |
inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
|
|
|
|
|
241 |
|
242 |
#Uncomment the following 4 lines, if you want LoRA Realism weights added to the pipeline
|
243 |
# inpaint_pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
|
|
|
238 |
|
239 |
#Setting up Flux (Schnell) Inpainting
|
240 |
inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
|
241 |
+
inpaint_pipe.load_lora_weights("XLabs-AI/flux-RealismLora")
|
242 |
+
|
243 |
|
244 |
#Uncomment the following 4 lines, if you want LoRA Realism weights added to the pipeline
|
245 |
# inpaint_pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
|