mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-01 19:49:03 +02:00
[to #42322933] add test condition for GPU availablity
damo/cv_diffusion_image-segmentation不支持cpu环境测试 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11689517
This commit is contained in:
@@ -1,6 +1,8 @@
|
||||
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||||
import unittest
|
||||
|
||||
import torch
|
||||
|
||||
from modelscope.pipelines import pipeline
|
||||
from modelscope.utils.constant import Tasks
|
||||
from modelscope.utils.demo_utils import DemoCompatibilityCheck
|
||||
@@ -15,6 +17,7 @@ class DDPMImageSemanticSegmentationTest(unittest.TestCase,
|
||||
self.model_id = 'damo/cv_diffusion_image-segmentation'
|
||||
|
||||
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||||
@unittest.skipIf(not torch.cuda.is_available(), 'cuda unittest')
|
||||
def test_ddpm_image_semantic_segmentation(self):
|
||||
input_location = 'data/test/images/image_ffhq34_00041527.png'
|
||||
|
||||
|
||||
Reference in New Issue
Block a user