Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 20:49:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix build break

Browse Source
This commit is contained in:
Yingda Chen
2022-12-13 12:45:58 -08:00
parent 93f2c52303
commit 0bf94e5c2d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/pipelines/cv/face_detection_pipeline.py
View File

@@ -34,7 +34,7 @@ class FaceDetectionPipeline(Pipeline):
config_path = osp.join(model, ModelFile.CONFIGURATION)
cfg = Config.from_file(config_path)
cfg_model = getattr(cfg, 'model', None)
if cfg_model is None:
if cfg_model is None or cfg_model.type == 'scrfd':
detector = ScrfdDetect(model_dir=model, **kwargs)
elif cfg_model.type == 'tinymog':
detector = self.model.to(self.device)