search

xinntao / facexlib

FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.
MIT License
810 stars 143 forks source link

can facexlib it support mac m1 architecture #24

Open ljc222 opened 1 year ago

ljc222 commented 1 year ago

hi, xinntal,Thank you for such an excellent project，i have a request,hope you can solve。

https://github.com/xinntao/facexlib/blob/master/facexlib/detection/retinaface.py in line 14 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') Can it support mac m1 architecture，use this code： device = torch.device('cuda' if torch.cuda.is_available() else 'mps' if torch.has_mps else 'cpu')