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

In-Context generation: allow lora composition

Browse Source
This commit is contained in:
Anthony Wu 2025-04-30 22:00:33 -07:00
parent bf977e80a0
commit 8b828ced27
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -85,7 +85,7 @@ class FluxInitializer:
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=lora_paths + hf_lora_paths,
lora_scales=lora_scales, lora_scales=lora_scales + [1.0] * len(hf_lora_paths),
) )
WeightHandlerLoRA.set_lora_weights( WeightHandlerLoRA.set_lora_weights(
transformer=flux_model.transformer, transformer=flux_model.transformer,

2
src/mflux/generate_in_context.py
View File

@ -25,7 +25,7 @@ def main():
quantize=args.quantize, quantize=args.quantize,
lora_names=[get_lora_filename(args.lora_style)] if args.lora_style else None, lora_names=[get_lora_filename(args.lora_style)] if args.lora_style else None,
lora_repo_id=LORA_REPO_ID if args.lora_style else None, lora_repo_id=LORA_REPO_ID if args.lora_style else None,
lora_paths=args.lora_paths if not args.lora_style else None, lora_paths=args.lora_paths,
lora_scales=args.lora_scales, lora_scales=args.lora_scales,
) )