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

adapt to cpu environment

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9754314
This commit is contained in:
biwen.lbw
2022-08-15 16:31:17 +08:00
committed by Yingda Chen
parent 77c6420c42
commit 4e0ee1ed57
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/pipelines/cv/skin_retouching_pipeline.py
View File

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