more pydantic fixes (#1338)

axolotl-ai-cloud · Feb 27, 2024 · 3f69571 · 3f69571
1 parent 1e3d530
commit 3f69571
Showing 1 changed file with 3 additions and 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
@@ -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