convert enum to str when referencing

Signed-off-by: Charlie Doern <[email protected]>
instructlab · Jun 25, 2024 · 700dad0 · 700dad0
1 parent ee76c17
commit 700dad0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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):
     """