I used ` cuda: "11.6.2"` and it seems to work. https://github.com/replicate/cog-stable-diffusion/blob/7a21fcd207cebbbf9e149b60cf9e9faa48c479ab/cog.yaml#L3