kjsman / stable-diffusion-pytorch

Yet another PyTorch implementation of Stable Diffusion (probably easy to read)
MIT License
522 stars 58 forks source link

Support running models in float16 #15

Closed rnxyfvls closed 6 months ago

rnxyfvls commented 6 months ago

To use float16, convert all models passed in to 'models' to float16

kjsman commented 6 months ago

Cool. Thank you for your contribution!