mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-19 12:12:48 +08:00
[hparams] fix data args (#8863)
This commit is contained in:
parent
df364998b1
commit
b2395b25b0
@ -19,6 +19,7 @@ sentencepiece
|
|||||||
tiktoken
|
tiktoken
|
||||||
modelscope>=1.14.0
|
modelscope>=1.14.0
|
||||||
hf-transfer
|
hf-transfer
|
||||||
|
safetensors<=0.5.3
|
||||||
# python
|
# python
|
||||||
fire
|
fire
|
||||||
omegaconf
|
omegaconf
|
||||||
|
@ -35,7 +35,7 @@ class DataArguments:
|
|||||||
default=None,
|
default=None,
|
||||||
metadata={"help": "The name of dataset(s) to use for evaluation. Use commas to separate multiple datasets."},
|
metadata={"help": "The name of dataset(s) to use for evaluation. Use commas to separate multiple datasets."},
|
||||||
)
|
)
|
||||||
dataset_dir: Union[str, dict] = field(
|
dataset_dir: str = field(
|
||||||
default="data",
|
default="data",
|
||||||
metadata={"help": "Path to the folder containing the datasets."},
|
metadata={"help": "Path to the folder containing the datasets."},
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user