yichang.zyc yingda.chen 3 years ago
parent
commit
5cbd9ff8fd
3 changed files with 3 additions and 3 deletions
  1. +1
    -1
      modelscope/pipelines/builder.py
  2. +1
    -1
      tests/pipelines/test_multi_modal_embedding.py
  3. +1
    -1
      tests/trainers/test_clip_multi_modal_embedding_trainer.py

+ 1
- 1
modelscope/pipelines/builder.py View File

@@ -74,7 +74,7 @@ DEFAULT_MODEL_FOR_PIPELINE = {
'damo/cv_resnet50_video-category'),
Tasks.multi_modal_embedding:
(Pipelines.multi_modal_embedding,
'damo/multi-modal_clip-vit-large-patch14-chinese_multi-modal-embedding'),
'damo/multi-modal_clip-vit-large-patch14_zh'),
Tasks.generative_multi_modal_embedding:
(Pipelines.generative_multi_modal_embedding,
'damo/multi-modal_gemm-vit-large-patch14_generative-multi-modal-embedding'


+ 1
- 1
tests/pipelines/test_multi_modal_embedding.py View File

@@ -11,7 +11,7 @@ from modelscope.utils.test_utils import test_level


class MultiModalEmbeddingTest(unittest.TestCase):
model_id = 'damo/multi-modal_clip-vit-large-patch14-chinese_multi-modal-embedding'
model_id = 'damo/multi-modal_clip-vit-large-patch14_zh'
test_text = {'text': '一张风景图'}

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')


+ 1
- 1
tests/trainers/test_clip_multi_modal_embedding_trainer.py View File

@@ -24,7 +24,7 @@ def clip_train_worker(local_rank, ngpus, node_size, node_rank):
dist.init_process_group(
backend='nccl', world_size=dist_world_size, rank=global_rank)

model_id = 'damo/multi-modal_clip-vit-large-patch14-chinese_multi-modal-embedding'
model_id = 'damo/multi-modal_clip-vit-large-patch14_zh'
local_model_dir = snapshot_download(model_id)

default_args = dict(


Loading…
Cancel
Save