From 700dad0cfc485d7b55daa57a46180850350d7097 Mon Sep 17 00:00:00 2001 From: Charlie Doern Date: Mon, 24 Jun 2024 23:34:50 -0400 Subject: [PATCH] convert enum to str when referencing Signed-off-by: Charlie Doern --- src/instructlab/training/config.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/instructlab/training/config.py b/src/instructlab/training/config.py index 83c7a1f8..69a61910 100644 --- a/src/instructlab/training/config.py +++ b/src/instructlab/training/config.py @@ -74,8 +74,12 @@ class LoraOptions(BaseModel): target_modules: list[str] = Field( default_factory=lambda: ["q_proj", "k_proj", "v_proj", "o_proj"] ) + quantize_data_type: QuantizeDataType = QuantizeDataType.NONE + class Config: + use_enum_values = True + class DeepSpeedOptions(BaseModel): """