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'), 'damo/cv_resnet50_video-category'),
Tasks.multi_modal_embedding: Tasks.multi_modal_embedding:
(Pipelines.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: Tasks.generative_multi_modal_embedding:
(Pipelines.generative_multi_modal_embedding, (Pipelines.generative_multi_modal_embedding,
'damo/multi-modal_gemm-vit-large-patch14_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): 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': '一张风景图'} test_text = {'text': '一张风景图'}


@unittest.skipUnless(test_level() >= 2, 'skip test in current test level') @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( dist.init_process_group(
backend='nccl', world_size=dist_world_size, rank=global_rank) 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) local_model_dir = snapshot_download(model_id)


default_args = dict( default_args = dict(


Loading…
Cancel
Save