From dac2167b7a8ed88147f4cc699970c73b6d742c02 Mon Sep 17 00:00:00 2001 From: "biwen.lbw" Date: Mon, 15 Aug 2022 16:31:17 +0800 Subject: [PATCH] adapt to cpu environment Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9754314 --- modelscope/pipelines/cv/skin_retouching_pipeline.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/cv/skin_retouching_pipeline.py b/modelscope/pipelines/cv/skin_retouching_pipeline.py index 056409df..d9b49ff3 100644 --- a/modelscope/pipelines/cv/skin_retouching_pipeline.py +++ b/modelscope/pipelines/cv/skin_retouching_pipeline.py @@ -44,8 +44,10 @@ class SkinRetouchingPipeline(Pipeline): """ super().__init__(model=model) - if device == 'gpu': + if torch.cuda.is_available() and device == 'gpu': device = 'cuda' + else: + device = 'cpu' model_path = os.path.join(self.model, ModelFile.TORCH_MODEL_FILE) detector_model_path = os.path.join( self.model, 'retinaface_resnet50_2020-07-20_old_torch.pth')