Remove redundant is_mflux weight attribute
This commit is contained in:
parent
696762af2e
commit
273431f010
@ -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
|
||||
|
||||
|
||||
@ -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
|
||||
|
||||
|
||||
|
||||
@ -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)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user