monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small fix for LoRA scales

Browse Source
This commit is contained in:
filipstrand 2025-05-06 13:01:26 +02:00
parent 63bf8710d1
commit 393a60fb2b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/mflux/flux/flux_initializer.py
View File

@ -33,8 +33,6 @@ class FluxInitializer:
# 0. Set paths, configs, and prompt_cache for later # 0. Set paths, configs, and prompt_cache for later
lora_paths = lora_paths or [] lora_paths = lora_paths or []
flux_model.prompt_cache = {} flux_model.prompt_cache = {}
flux_model.lora_paths = lora_paths
flux_model.lora_scales = lora_scales
flux_model.model_config = model_config flux_model.model_config = model_config
# 1. Initialize tokenizers # 1. Initialize tokenizers
@ -82,10 +80,12 @@ class FluxInitializer:
lora_names=lora_names, lora_names=lora_names,
repo_id=lora_repo_id, repo_id=lora_repo_id,
) )
flux_model.lora_paths = lora_paths + hf_lora_paths
flux_model.lora_scales = (lora_scales or []) + [1.0] * len(hf_lora_paths)
lora_weights = WeightHandlerLoRA.load_lora_weights( lora_weights = WeightHandlerLoRA.load_lora_weights(
transformer=flux_model.transformer, transformer=flux_model.transformer,
lora_files=lora_paths + hf_lora_paths, lora_files=flux_model.lora_paths,
lora_scales=lora_scales + [1.0] * len(hf_lora_paths), lora_scales=flux_model.lora_scales,
) )
WeightHandlerLoRA.set_lora_weights( WeightHandlerLoRA.set_lora_weights(
transformer=flux_model.transformer, transformer=flux_model.transformer,