Not able to run on CPU - Githubissues

HazyResearch / domino

Apache License 2.0

134 stars 24 forks source link

Not able to run on CPU #2

Closed sasha-scale closed 2 years ago

sasha-scale commented 2 years ago

Some of the code in the repository assumes access to a gpu device, i.e. model.to(0) - would probably be good for end users if both CPU and GPU were supported

seyuboglu commented 2 years ago

Good point – just pushed some changes that got rid of any explicit to(0) in the code and notebooks.