Former-commit-id: 3af57795dd
This commit is contained in:
hiyouga
2024-10-11 23:51:54 +08:00
parent 012f4fef6b
commit e90a1199da
12 changed files with 91 additions and 69 deletions

View File

@@ -109,15 +109,15 @@ def get_model_path(model_name: str) -> str:
use_modelscope()
and path_dict.get(DownloadSource.MODELSCOPE)
and model_path == path_dict.get(DownloadSource.DEFAULT)
): # replace path
): # replace hf path with ms path
model_path = path_dict.get(DownloadSource.MODELSCOPE)
if (
use_openmind()
and path_dict.get(DownloadSource.MODELERS)
and path_dict.get(DownloadSource.OPENMIND)
and model_path == path_dict.get(DownloadSource.DEFAULT)
): # replace path
model_path = path_dict.get(DownloadSource.MODELERS)
): # replace hf path with om path
model_path = path_dict.get(DownloadSource.OPENMIND)
return model_path