From d5f5f9607e0dd36f6a9b32b79dffec9a2d0e3299 Mon Sep 17 00:00:00 2001 From: Filip Strand Date: Wed, 4 Sep 2024 23:14:38 +0200 Subject: [PATCH] Fix type Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com> --- src/flux_1/flux.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/flux_1/flux.py b/src/flux_1/flux.py index 92aa750..bbae938 100644 --- a/src/flux_1/flux.py +++ b/src/flux_1/flux.py @@ -29,7 +29,7 @@ class Flux1: quantize_full_weights: int | None = None, local_path: str | None = None, lora_path: str | None = None, - lora_scale: str | None = None, + lora_scale: float = 1.0, ): self.model_config = model_config self.quantize_full_weights = quantize_full_weights