Former-commit-id: 1fe424323b212094856f423351dc2a15774d39c3
This commit is contained in:
hiyouga
2024-10-11 23:51:54 +08:00
parent c71c78da50
commit c7efc7f2ed
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