• brkirch's avatar
    Add option for float32 sampling with float16 UNet · 84d9ce30
    brkirch authored
    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().
    84d9ce30
Name
Last commit
Last update
.github Loading commit data...
configs Loading commit data...
embeddings Loading commit data...
extensions Loading commit data...
extensions-builtin Loading commit data...
html Loading commit data...
javascript Loading commit data...
localizations Loading commit data...
models Loading commit data...
modules Loading commit data...
scripts Loading commit data...
test Loading commit data...
textual_inversion_templates Loading commit data...
.gitignore Loading commit data...
.pylintrc Loading commit data...
CODEOWNERS Loading commit data...
LICENSE.txt Loading commit data...
README.md Loading commit data...
environment-wsl2.yaml Loading commit data...
launch.py Loading commit data...
requirements.txt Loading commit data...
requirements_versions.txt Loading commit data...
screenshot.png Loading commit data...
script.js Loading commit data...
style.css Loading commit data...
v2-inference-v.yaml Loading commit data...
webui-macos-env.sh Loading commit data...
webui-user.bat Loading commit data...
webui-user.sh Loading commit data...
webui.bat Loading commit data...
webui.py Loading commit data...
webui.sh Loading commit data...