Remove redundant is_mflux weight attribute

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

View File

@ -58,7 +58,7 @@ class LoRALayers:
lora_layers = {**transformer_lora_layers, **single_transformer_lora_layers}
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'],
) # fmt:off

View File

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

View File

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