[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:
shuying.shu
2023-02-16 11:13:53 +00:00
committed by wenmeng.zwm
parent b99a8a94e6
commit 615c443020

View File

@@ -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'