unexpected error, argument #3 'index' in call to _th_index_select

[javismiles]

Im getting this error when following the readme instructions: RuntimeError: Expected object of device type cuda but got device type cpu for argument #3 'index' in call to _th_index_select

any tips?

(im running it in colab with cuda GPU)