Commit
·
ecfb7d9
1
Parent(s):
7733adf
Refactor flux_helpers.py to fix loading of Lora weights
Browse files
modules/helpers/flux_helpers.py
CHANGED
|
@@ -60,7 +60,7 @@ def get_pipe(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
|
|
| 60 |
# Set Loras
|
| 61 |
if request.loras:
|
| 62 |
for i, lora in enumerate(request.loras):
|
| 63 |
-
pipe_args["pipeline"].load_lora_weights(
|
| 64 |
adapter_names = [f"lora_{i}" for i in range(len(request.loras))]
|
| 65 |
adapter_weights = [lora['weight'] for lora in request.loras]
|
| 66 |
|
|
|
|
| 60 |
# Set Loras
|
| 61 |
if request.loras:
|
| 62 |
for i, lora in enumerate(request.loras):
|
| 63 |
+
pipe_args["pipeline"].load_lora_weights(lora['repo_id'], adapter_name=f"lora_{i}")
|
| 64 |
adapter_names = [f"lora_{i}" for i in range(len(request.loras))]
|
| 65 |
adapter_weights = [lora['weight'] for lora in request.loras]
|
| 66 |
|