version: 1.6.1

README.md

@@ -167,6 +167,8 @@ docker run -it --gpus all -v $HOME/.cache/huggingface:/root/.cache/huggingface -
 [Example Colab](https://colab.research.google.com/drive/1rOvQNs0Cmn_yU1bKWjCOHzGVDgZkaTtO?usp=sharing)
 
 ## ChangeLog
+**1.6.1**
+ - fix: make sure image tensors come to the CPU as float32 so there aren't compatability issues with non-bfloat16 cpus
 
 **1.6.0**
  - fix: *maybe* address #13 with `expected scalar type BFloat16 but found Float`

setup.py

@@ -7,7 +7,7 @@ setup(
     name="imaginAIry",
     author="Bryce Drennan",
     # author_email="b r y p y d o t io",
-    version="1.6.0",
+    version="1.6.1",
     description="AI imagined images. Pythonic generation of stable diffusion images.",
     long_description=readme,
     long_description_content_type="text/markdown",