Add model download to train_ms.py (#150)

* download base model * fix download * fix download repo * set HF default
2023-11-06 06:42:07 +08:00
parent 0dbe362c83
commit 80c418e807
4 changed files with 51 additions and 2 deletions
--- a/config.py
+++ b/config.py
@@ -118,9 +118,11 @@ class Train_ms_config:
        self,
        config_path: str,
        env: Dict[str, any],
+        base: Dict[str, any],
        model: str,
    ):
        self.env = env  # 需要加载的环境变量
+        self.base = base  # 底模配置
        self.model = model  # 训练模型存储目录，该路径为相对于dataset_path的路径，而非项目根目录
        self.config_path = config_path  # 配置文件路径