Fix type

Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
2024-09-04 23:14:38 +02:00 · 2024-09-04 23:14:38 +02:00 · d5f5f9607e
commit d5f5f9607e
parent f28fffe0cf
1 changed files with 1 additions and 1 deletions
--- 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