diff --git a/src/instructlab/training/config.py b/src/instructlab/training/config.py index 24da6e95..db50c397 100644 --- a/src/instructlab/training/config.py +++ b/src/instructlab/training/config.py @@ -69,7 +69,7 @@ class LoraOptions(BaseModel): """ rank: int = 4 - alpha: float = 32 + alpha: int = 32 dropout: float = 0.1 target_modules: list[str] = Field( default_factory=lambda: ["q_proj", "k_proj", "v_proj", "o_proj"]