mirror of
https://github.com/Stability-AI/stablediffusion.git
synced 2024-12-22 15:44:58 +00:00
506403501f
In the `paint()` function, you are using the `torch.autocast()` function to cast the batch tensor to the `cuda` device. However, the `batch` tensor is already on the `cuda` device, so there is no need to cast it again. This can be fixed by removing the `torch.autocast()` call from the `paint()` function. |
||
---|---|---|
.. | ||
gradio | ||
streamlit | ||
tests | ||
img2img.py | ||
txt2img.py |