tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

more pydantic fixes (#1338)

Browse Source
This commit is contained in:
Wing Lian
2024-02-26 22:39:13 -05:00
committed by GitHub
parent 1e3d5305d3
commit 3f69571943
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
View File

@@ -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