Closed alexhmyang closed 1 year ago
Hi @alexhmyang , to load the weight you need to have GPU on your computer and more specifically, installed CUDA on it. If you don't have GPU then you should use my code on google colab. Sincere!
thanks! where is colab code ? I havent fount it
Here is the link, you can try it, but I saw that it has some issue with downloading the weights. So I will try to solve it asap!
i had a similar issue, chatgpt helped a bit but what worked for me is changing cuda:0 with cpu in the device parameter in the 3 .yaml files in in the config folder
Yes, I forgot that I have included a device parameter in the config files. But I don't know if it works anymore in this version. Feel free to play with those configs and contact me if any issue happens. Thanks!
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