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

Remove redundant is_mflux weight attribute

Browse Source
This commit is contained in:
filipstrand 2025-03-15 10:22:36 +01:00
parent 696762af2e
commit 273431f010
3 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/mflux/dreambooth/lora_layers/lora_layers.py
View File

@ -58,7 +58,7 @@ class LoRALayers:
lora_layers = {**transformer_lora_layers, **single_transformer_lora_layers} lora_layers = {**transformer_lora_layers, **single_transformer_lora_layers}
weights = WeightHandler( weights = WeightHandler(
meta_data=MetaData(is_mflux=True), meta_data=MetaData(mflux_version=GeneratedImage.get_version()),
transformer=mlx.utils.tree_unflatten(list(lora_layers.items()))['transformer'], transformer=mlx.utils.tree_unflatten(list(lora_layers.items()))['transformer'],
) # fmt:off ) # fmt:off

1
src/mflux/weights/weight_handler.py
View File

@ -14,7 +14,6 @@ class MetaData:
quantization_level: int | None = None quantization_level: int | None = None
scale: float | None = None scale: float | None = None
is_lora: bool = False is_lora: bool = False
is_mflux: bool = False
mflux_version: str | None = None mflux_version: str | None = None

2
src/mflux/weights/weight_handler_lora.py
View File

@ -38,7 +38,7 @@ class WeightHandlerLoRA:
quantization_level=None, quantization_level=None,
scale=lora_scale, scale=lora_scale,
is_lora=True, is_lora=True,
is_mflux=True if mflux_version is not None else False, mflux_version=mflux_version,
), ),
) )
lora_weights.append(weights) lora_weights.append(weights)