RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select)

I’m loading model in google colab w. automatic1111 UI
and when I use model that aren’t AOM, I get this error
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select)

here’s my colab notebook:

Hey @Pikelz :wave:

The information you shared is not enough to pinpoint the problem. However, since your collab is based on GitHub - camenduru/stable-diffusion-webui: Stable Diffusion web UI, have you tried opening an issue there? :slight_smile:

bump - running into same issue - lmk if you found any solutions pls!