Browse Source

[to #46408569]fix: pipeline and trainer user-agent add not replacement.

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10890156

    * [to #46408569]fix: pipeline and trainer user-agent add not replacement.
master^2
mulin.lyh 3 years ago
parent
commit
2a8e653169
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      modelscope/models/base/base_model.py

+ 2
- 2
modelscope/models/base/base_model.py View File

@@ -107,9 +107,9 @@ class Model(ABC):
)

if invoked_by is not None:
invoked_by = {Invoke.KEY: invoked_by}
invoked_by = '%s/%s' % (Invoke.KEY, invoked_by)
else:
invoked_by = {Invoke.KEY: Invoke.PRETRAINED}
invoked_by = '%s/%s' % (Invoke.KEY, Invoke.PRETRAINED)
local_model_dir = snapshot_download(
model_name_or_path, revision, user_agent=invoked_by)
logger.info(f'initialize model from {local_model_dir}')


Loading…
Cancel
Save