Skip to content

Input type (MPSFloatType) and weight type (torch.FloatTensor) should be the same #3

@lunatolun

Description

@lunatolun

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions