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

fix ut error

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13716099
* fix ut error
This commit is contained in:
wenmeng.zwm
2023-08-19 21:16:07 +08:00
parent 1c76007da7
commit 7f7573f15d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/pipelines/test_text_to_360panorama_image.py
View File

@@ -9,7 +9,6 @@ import cv2
from modelscope.hub.snapshot_download import snapshot_download
from modelscope.outputs import OutputKeys
from modelscope.pipelines import pipeline
from modelscope.pipelines.cv import Text2360PanoramaImagePipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.logger import get_logger
from modelscope.utils.test_utils import test_level
@@ -41,8 +40,9 @@ class Text2360PanoramaImageTest(unittest.TestCase):
'refinement': self.refinement,
}
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 3, 'skip test due to gpu oom')
def test_run_by_direct_model_download(self):
from modelscope.pipelines.cv import Text2360PanoramaImagePipeline
output_image_path = tempfile.NamedTemporaryFile(suffix='.png').name
cache_path = snapshot_download(self.model_id)
pipeline = Text2360PanoramaImagePipeline(cache_path)
@@ -52,8 +52,9 @@ class Text2360PanoramaImageTest(unittest.TestCase):
print(
'pipeline: the output image path is {}'.format(output_image_path))
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 3, 'skip test due to gpu oom')
def test_run_with_model_from_modelhub(self):
from modelscope.pipelines.cv import Text2360PanoramaImagePipeline
output_image_path = tempfile.NamedTemporaryFile(suffix='.png').name
pipeline_ins = pipeline(
task=Tasks.text_to_360panorama_image,