add readme

Former-commit-id: 3d5ec6f12b4ae7d04520e6865516a9a6dd4f7efe
This commit is contained in:
yuze.zyz
2023-12-01 16:11:30 +08:00
parent 8547085615
commit d323ccc3ec
4 changed files with 61 additions and 3 deletions

View File

@@ -43,7 +43,7 @@ def register_model_group(
else:
assert prefix == name.split("-")[0], "prefix should be identical."
if not os.environ.get('USE_MODELSCOPE_HUB', False):
if not int(os.environ.get('USE_MODELSCOPE_HUB', '0')):
# If path is a string, we treat it as a huggingface model-id by default.
SUPPORTED_MODELS[name] = path["hf"] if isinstance(path, dict) else path
elif isinstance(path, dict) and "ms" in path:

View File

@@ -235,7 +235,7 @@ def load_model_and_tokenizer(
def try_download_model_from_ms(model_args):
if os.environ.get('USE_MODELSCOPE_HUB', False) and not os.path.exists(model_args.model_name_or_path):
if int(os.environ.get('USE_MODELSCOPE_HUB', '0')) and not os.path.exists(model_args.model_name_or_path):
try:
from modelscope import snapshot_download
revision = model_args.model_revision
@@ -243,5 +243,5 @@ def try_download_model_from_ms(model_args):
revision = 'master'
model_args.model_name_or_path = snapshot_download(model_args.model_name_or_path, revision)
except ImportError as e:
raise ImportError(f'You are using `USE_MODELSCOPE_HUB=True` but you have no modelscope sdk installed. '
raise ImportError(f'You are using `USE_MODELSCOPE_HUB=1` but you have no modelscope sdk installed. '
f'Please install it by `pip install modelscope -U`') from e