From 3f695719431cc095aac2278227d8525719c20b85 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Mon, 26 Feb 2024 22:39:13 -0500
Subject: [PATCH] more pydantic fixes (#1338)

---
 src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
index 3d9b54e566..062f60a6e8 100644
--- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
+++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
@@ -76,7 +76,7 @@ class SFTDataset(BaseModel):
     type: Optional[Union[str, UserDefinedPrompterType]] = None
     shards: Optional[int] = None
     conversation: Optional[str] = None
-    data_files: Optional[List[str]] = None
+    data_files: Optional[Union[str, List[str]]] = None
     name: Optional[str] = None
     ds_type: Optional[str] = None
     train_on_split: Optional[str] = None
@@ -454,9 +454,9 @@ class AxolotlInputConfig(
     max_steps: Optional[int] = None
     warmup_steps: Optional[int] = None
     warmup_ratio: Optional[float] = None
-    eval_steps: Optional[int] = None
+    eval_steps: Optional[Union[int, float]] = None
     evaluation_strategy: Optional[str] = None
-    save_steps: Optional[int] = None
+    save_steps: Optional[Union[int, float]] = None
     saves_per_epoch: Optional[int] = None
     save_strategy: Optional[str] = None
     save_total_limit: Optional[int] = None