NikhilJoson commited on
Commit
59d5e77
·
verified ·
1 Parent(s): 32e67bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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")