barreloflube commited on
Commit
ecfb7d9
·
1 Parent(s): 7733adf

Refactor flux_helpers.py to fix loading of Lora weights

Browse files
Files changed (1) hide show
  1. modules/helpers/flux_helpers.py +1 -1
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(request.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
 
 
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