feat(tools/train): 优化训练参数传递

2025-11-30 13:49:24 +08:00
parent 3ee84b31a0
commit db53cc5001
2 changed files with 39 additions and 41 deletions
@@ -249,10 +249,9 @@ class DatasetLoader:
    @staticmethod       
    def load(
        train_type: Literal["seq", "sft", "dpo"],
-        load_path: Union[str, List[str]],
+        load_path: str,
        window_size: int, 
        stride: Optional[int] = None,
-        **kwargs
        ) -> BaseDataset:
        if stride is None:
            stride = window_size