diff --git a/README.md b/README.md index d47a23a..0bb24a4 100644 --- a/README.md +++ b/README.md @@ -85,7 +85,7 @@ python main.py --model dev --prompt "Luxury food photograph" --steps 25 --seed 2 - **`--apply-lora`** (optional, `[str]`, default: `[]`): [Lora Safetensors file] -- **`--lora-scales`** (optional, `[float]`, default: `[1.0]`): [Scaling factor for each LoRA files] +- **`--lora-scale`** (optional, `[float]`, default: `[1.0]`): [Scaling factor for each LoRA files] ### Note: Ensure that the safetensors file provided is compatible with the model's architecture and that the LoRA keys correctly map to the model's layers. The missing lora-scales shall be treated as 1.0 by default. diff --git a/src/flux_1/weights/weight_handler.py b/src/flux_1/weights/weight_handler.py index bc6669e..fbc2a88 100644 --- a/src/flux_1/weights/weight_handler.py +++ b/src/flux_1/weights/weight_handler.py @@ -15,7 +15,7 @@ class WeightHandler: repo_id: str | None = None, local_path: str | None = None, lora_path: str | None = None, - lora_scale: float | None = None, + lora_scale: float = 1.0, ): root_path = Path(local_path) if local_path else WeightHandler._download_or_get_cached_weights(repo_id)