automatic

Commit Graph

Author	SHA1	Message	Date
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
AUTOMATIC	c81d440d87	moved deepdanbooru to pure pytorch implementation	2022-11-20 16:39:20 +03:00

Author

SHA1

Message

Date

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

AUTOMATIC

c81d440d87

moved deepdanbooru to pure pytorch implementation

2022-11-20 16:39:20 +03:00

2 Commits (2dabba919052d8c98dee8d6725ba43de89016691)