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().
显示
- README.md 1 个添加, 0 个删除README.md
- modules/deepbooru_model.py 3 个添加, 1 个删除modules/deepbooru_model.py
- modules/devices.py 2 个添加, 0 个删除modules/devices.py
- modules/processing.py 8 个添加, 7 个删除modules/processing.py
- modules/sd_hijack_unet.py 29 个添加, 0 个删除modules/sd_hijack_unet.py
- modules/sd_hijack_utils.py 28 个添加, 0 个删除modules/sd_hijack_utils.py
- modules/sd_models.py 10 个添加, 0 个删除modules/sd_models.py
- modules/shared.py 1 个添加, 0 个删除modules/shared.py
加载中
想要评论请 注册 或 登录