From 2a8e6531692b247b73e9dcbd3c0293d8b779cccc Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Mon, 28 Nov 2022 19:45:58 +0800 Subject: [PATCH] [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. --- modelscope/models/base/base_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/models/base/base_model.py b/modelscope/models/base/base_model.py index 5f22b320..1f464bf3 100644 --- a/modelscope/models/base/base_model.py +++ b/modelscope/models/base/base_model.py @@ -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}')