-
Notifications
You must be signed in to change notification settings - Fork 10
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Inference doesn't run on cpu only device #16
Comments
Hi, we haven't tested on CPU-based inference or Google Colab, given the compute requirements. Your solution should be fine as a workaround. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Attempting to run
inference.run_inference()
withdevice='cpu'
on a Google colab CPU only instance produces an error.RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
I worked around this by commenting out lines 30 to 32 of /ldm/models/diffusion/ddim.py. I am not sure if this will cause problems elsewhere that I haven't noticed.
The text was updated successfully, but these errors were encountered: