codeformer-pip
codeformer-pip copied to clipboard
Input type (MPSFloatType) and weight type (torch.FloatTensor) should be the same
Hello, I am on M1 Mac, with MPS, and the solution is here in case somebody encounters the same issue (just a quick, dirty workaround) is go to site-packages/codeformer/app.py
replace line 71:
# REPLACE THIS LINE
# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
device = torch.device('mps')