diff --git a/modelscope/pipelines/builder.py b/modelscope/pipelines/builder.py index 346d8048..96043ce9 100644 --- a/modelscope/pipelines/builder.py +++ b/modelscope/pipelines/builder.py @@ -57,6 +57,9 @@ DEFAULT_MODEL_FOR_PIPELINE = { Tasks.visual_question_answering: (Pipelines.visual_question_answering, 'damo/mplug_visual-question-answering_coco_large_en'), + Tasks.text_to_image_synthesis: + (Pipelines.text_to_image_synthesis, + 'damo/cv_imagen_text-to-image-synthesis_tiny') } diff --git a/modelscope/pipelines/cv/action_recognition_pipeline.py b/modelscope/pipelines/cv/action_recognition_pipeline.py index 8eefa301..757f87e3 100644 --- a/modelscope/pipelines/cv/action_recognition_pipeline.py +++ b/modelscope/pipelines/cv/action_recognition_pipeline.py @@ -2,9 +2,6 @@ import math import os.path as osp from typing import Any, Dict -import cv2 -import numpy as np -import PIL import torch from modelscope.metainfo import Pipelines diff --git a/modelscope/pipelines/multi_modal/text_to_image_synthesis_pipeline.py b/modelscope/pipelines/multi_modal/text_to_image_synthesis_pipeline.py index edffe1f2..02a34428 100644 --- a/modelscope/pipelines/multi_modal/text_to_image_synthesis_pipeline.py +++ b/modelscope/pipelines/multi_modal/text_to_image_synthesis_pipeline.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, Union +from typing import Any, Dict from modelscope.metainfo import Pipelines from modelscope.pipelines.base import Input @@ -23,7 +23,7 @@ class TextToImageSynthesisPipeline(Pipeline): pipe_model = model else: raise NotImplementedError( - f'execpting a Model instance or str, but get {type(model)}.') + f'expecting a Model instance or str, but get {type(model)}.') super().__init__(model=pipe_model) diff --git a/tests/pipelines/test_action_recognition.py b/tests/pipelines/test_action_recognition.py index 6f608041..7453f136 100644 --- a/tests/pipelines/test_action_recognition.py +++ b/tests/pipelines/test_action_recognition.py @@ -1,14 +1,10 @@ # Copyright (c) Alibaba, Inc. and its affiliates. # !/usr/bin/env python import os.path as osp -import shutil import tempfile import unittest -import cv2 - from modelscope.fileio import File -from modelscope.msdatasets import MsDataset from modelscope.pipelines import pipeline from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.test_utils import test_level @@ -45,7 +41,7 @@ class ActionRecognitionTest(unittest.TestCase): print(f'recognition output: {result}.') - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') def test_run_modelhub_default_model(self): recognition_pipeline = pipeline(Tasks.action_recognition) result = recognition_pipeline(