From e27b84fe37c29218fd5e539a0e65717a73c17dfa Mon Sep 17 00:00:00 2001 From: "zhangzhicheng.zzc" Date: Wed, 12 Apr 2023 20:38:13 +0800 Subject: [PATCH] plugin add modelscope source Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/12306018 --- modelscope/utils/plugins.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/modelscope/utils/plugins.py b/modelscope/utils/plugins.py index bcc920d8..a83ca03c 100644 --- a/modelscope/utils/plugins.py +++ b/modelscope/utils/plugins.py @@ -635,6 +635,10 @@ class PluginsManager(object): Returns: """ + + if package.split('.')[-1] == 'whl': + return False, '' + from pip._internal.utils.packaging import get_requirement, specifiers req = get_requirement(package) @@ -675,6 +679,11 @@ class PluginsManager(object): """ from pip._internal.commands import create_command importlib.reload(pkg_resources) + if command == 'install': + command_args.append('-f') + command_args.append( + 'https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html' + ) command = create_command(command) options, args = command.parse_args(command_args)