load loras

app.py

@@ -121,6 +121,8 @@ def process(
     resized_image = image.resize((width, height), Image.LANCZOS)
     resized_mask = mask.resize((width, height), Image.LANCZOS)
 
+    pipe.load_lora_weights("jiuface/boy-001", weight_name="flux_train_replicate.safetensors")
+    lora_scale = 0.8
     if randomize_seed_checkbox:
         seed_slicer = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed_slicer)
@@ -132,7 +134,8 @@ def process(
         height=height,
         strength=strength_slider,
         generator=generator,
-        num_inference_steps=num_inference_steps_slider
+        num_inference_steps=num_inference_steps_slider,
+        joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
     print('INFERENCE DONE')
     return result, resized_mask