Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #46852519]fix: fix useragent string and a trainer invokedby

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11164288

    * [to #46852519]fix: fix useragent string and a trainer invokedby
This commit is contained in:
mulin.lyh
2022-12-21 08:28:40 +08:00
parent 02aa5f1e42
commit 9e8d7bed6a
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/hub/api.py
View File

@@ -809,7 +809,7 @@ class ModelScopeConfig:
user_name,
)
if isinstance(user_agent, dict):
ua += '; '.join(f'{k}/{v}' for k, v in user_agent.items())
ua += '; ' + '; '.join(f'{k}/{v}' for k, v in user_agent.items())
elif isinstance(user_agent, str):
ua += ';' + user_agent
ua += '; ' + user_agent
return ua

6
modelscope/trainers/cv/image_classifition_trainer.py
View File

@@ -18,7 +18,7 @@ from modelscope.msdatasets.ms_dataset import MsDataset
from modelscope.preprocessors.base import Preprocessor
from modelscope.trainers.base import BaseTrainer
from modelscope.trainers.builder import TRAINERS
from modelscope.utils.constant import DEFAULT_MODEL_REVISION, ModelFile
from modelscope.utils.constant import DEFAULT_MODEL_REVISION, Invoke, ModelFile
from modelscope.utils.logger import get_logger
@@ -202,7 +202,9 @@ class ImageClassifitionTrainer(BaseTrainer):
model) else os.path.dirname(model)
else:
self.model_dir = snapshot_download(
model, revision=model_revision)
model,
revision=model_revision,
user_agent={Invoke.KEY: Invoke.TRAINER})
if cfg_file is None:
cfg_file = os.path.join(self.model_dir,
ModelFile.CONFIGURATION)