| | |
| | | |
| | | def download_model(**kwargs): |
| | | hub = kwargs.get("hub", "ms") |
| | | if hub == "ms": |
| | | if hub == "ms" or hub == "modelscope": |
| | | kwargs = download_from_ms(**kwargs) |
| | | elif hub == "hf": |
| | | elif hub == "hf" or hub == "huggingface": |
| | | kwargs = download_from_hf(**kwargs) |
| | | elif hub == "openai": |
| | | model_or_path = kwargs.get("model") |
| | |
| | | if isinstance(file_path_metas, dict): |
| | | if isinstance(cfg, list): |
| | | cfg.append({}) |
| | | |
| | | for k, v in file_path_metas.items(): |
| | | if isinstance(v, str): |
| | | p = os.path.join(model_or_path, v) |
| | |
| | | if k not in cfg: |
| | | cfg[k] = [] |
| | | if isinstance(vv, str): |
| | | p = os.path.join(model_or_path, v) |
| | | file_path_metas[i] = p |
| | | p = os.path.join(model_or_path, vv) |
| | | # file_path_metas[i] = p |
| | | if os.path.exists(p): |
| | | if isinstance(cfg[k], dict): |
| | | cfg[k] = p |