support ms

Former-commit-id: fdd4f94f563110ef9f96ab4a7fd954def32e9785
2023-11-29 20:36:55 +08:00
parent 08d5340bd8
commit 9d125bf533
3 changed files with 188 additions and 1 deletions
--- a/src/llmtuner/hparams/model_args.py
+++ b/src/llmtuner/hparams/model_args.py
@@ -8,7 +8,8 @@ class ModelArguments:
    Arguments pertaining to which model/config/tokenizer we are going to fine-tune.
    """
    model_name_or_path: str = field(
-        metadata={"help": "Path to pretrained model or model identifier from huggingface.co/models."}
+        metadata={"help": "Path to pretrained model or model identifier "
+                          "from huggingface.co/models or modelscope.cn/models."}
    )
    cache_dir: Optional[str] = field(
        default=None,