From 08441fed17968fc92c901f1ef71934942316f634 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Mon, 23 Feb 2026 23:39:53 +0700 Subject: [PATCH] fix: set allowed values for `adapter` config (#3415) --- src/axolotl/utils/schemas/peft.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/axolotl/utils/schemas/peft.py b/src/axolotl/utils/schemas/peft.py index a9ce1fbd6..a86de7822 100644 --- a/src/axolotl/utils/schemas/peft.py +++ b/src/axolotl/utils/schemas/peft.py @@ -1,6 +1,6 @@ """Pydantic models for PEFT-related configuration""" -from typing import Any +from typing import Any, Literal from pydantic import BaseModel, Field, field_validator, model_validator @@ -38,10 +38,10 @@ class LoraConfig(BaseModel): default=False, json_schema_extra={"description": "Use bitsandbytes 4 bit"} ) - adapter: str | None = Field( + adapter: Literal["lora", "qlora", "llama-adapter"] | None = Field( default=None, json_schema_extra={ - "description": "If you want to use 'lora' or 'qlora' or leave blank to train all parameters in original model" + "description": "If you want to use 'lora', 'qlora', or 'llama-adapter', or leave blank to train all parameters in original model" }, ) lora_model_dir: str | None = Field(