I keep getting this issue, how do I fix it?

facebookresearch / ToMe

A method to increase the speed and lower the memory footprint of existing vision transformers.

Other

931 stars 67 forks source link

I keep getting this issue, how do I fix it? #26

Closed 303Aki303 closed 1 year ago

303Aki303 commented 1 year ago

raise NotImplementedError(f"Invalid/unsupported device. Expected cpu, cuda, or mps, got {device.type}.") NotImplementedError: Invalid/unsupported device. Expected cpu, cuda, or mps, got privateuseone.

dbolya commented 1 year ago

I believe you opened this issue in the wrong github. That line of code only appears in tomesd. Can you move this issue over there. Thanks.