How to run this repo on cpu with autocast enabled?

Stability-AI / generative-models

Generative Models by Stability AI

MIT License

23.14k stars 2.56k forks source link

Open SKRohit opened 2 weeks ago

SKRohit commented 2 weeks ago

I want to run the model on the CPU for bf16 precision. Want to understand how to enable autocasting for the CPU?