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 |
|