AttributeError: Can't get attribute 'StemBlock' on <module 'models.common' from '/mnt/data/train_codes/yolov7-face-master/models/common.py'>

[xinlin-xiao]

use:python3 detect.py --weights ./weights/yolov7-lite-s.pt --source /mnt/data/train_codes/yolov7-face-master/data/images
return: return super().find_class(mod_name, name) AttributeError: Can't get attribute 'StemBlock' on <module 'models.common' from '/mnt/data/train_codes/yolov7-face-master/models/common.py'>

[xinlin-xiao]

use:yolov5-facehttps://github.com/deepcam-cn/yolov5-face/blob/master/models/common.py
return: return super().find_class(mod_name, name) AttributeError: Can't get attribute 'Shuffle_Block' on <module 'models.common' from '/mnt/data/train_codes/yolov7-face-master/models/common.py'> anyone can help me to fix it