brkirch 84d9ce30cb Add option for float32 sampling with float16 UNet ...

This also handles type casting so that ROCm and MPS torch devices work correctly without --no-half. One cast is required for deepbooru in deepbooru_model.py, some explicit casting is required for img2img and inpainting. depth_model can't be converted to float16 or it won't work correctly on some systems (it's known to have issues on MPS) so in sd_models.py model.depth_model is removed for model.half().

2023-01-25 01:13:02 -05:00

36 KiB

Raw Permalink Blame History

View Raw