How to run on CUDA? - Githubissues

divamgupta / stable-diffusion-tensorflow

Stable Diffusion in TensorFlow / Keras

Other

1.57k stars 227 forks source link

How to run on CUDA? #77

Open boredcoder411 opened 8 months ago

boredcoder411 commented 8 months ago

I can't seem to figure out how to move the model to my GPU. I've tried copying the cloab notebooks, to no avail. Opening task manager shows the code is running on my cpu.