aboutsummaryrefslogtreecommitdiffstats
path: root/modules/deepbooru_model.py
Commit message (Collapse)AuthorAgeFilesLines
* Add option for float32 sampling with float16 UNetbrkirch2023-01-251-1/+3
| | | | 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().
* moved deepdanbooru to pure pytorch implementationAUTOMATIC2022-11-201-0/+676