mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 03:32:50 +08:00
add om_hub_token argument
Former-commit-id: 7b91be33c9cd8473453716f0c4c6dec924304efc
This commit is contained in:
parent
132c1f1b0f
commit
e8e98bb125
@ -96,7 +96,7 @@ def _load_single_dataset(
|
||||
data_files=data_files,
|
||||
split=dataset_attr.split,
|
||||
cache_dir=cache_dir,
|
||||
token=model_args.ms_hub_token,
|
||||
token=model_args.om_hub_token,
|
||||
streaming=(data_args.streaming and (dataset_attr.load_from != "file")),
|
||||
)
|
||||
except ImportError:
|
||||
|
@ -563,7 +563,7 @@ register_model_group(
|
||||
"Gemma-2-9B-Instruct": {
|
||||
DownloadSource.DEFAULT: "google/gemma-2-9b-it",
|
||||
DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-9b-it",
|
||||
DownloadSource.MODELERS: "LlamaFactory/Qwen2-VL-2B-Instruct"
|
||||
DownloadSource.MODELERS: "LlamaFactory/gemma-2-2b-it"
|
||||
},
|
||||
"Gemma-2-27B-Instruct": {
|
||||
DownloadSource.DEFAULT: "google/gemma-2-27b-it",
|
||||
|
@ -267,6 +267,10 @@ class ModelArguments(QuantizationArguments, ProcessorArguments, ExportArguments,
|
||||
default=None,
|
||||
metadata={"help": "Auth token to log in with ModelScope Hub."},
|
||||
)
|
||||
om_hub_token: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "Auth token to log in with Modelers Hub."},
|
||||
)
|
||||
print_param_status: bool = field(
|
||||
default=False,
|
||||
metadata={"help": "For debugging purposes, print the status of the parameters in the model."},
|
||||
|
Loading…
x
Reference in New Issue
Block a user