Sept. 18, 2022, 2:34 p.m. | /u/hardmaru

Machine Learning www.reddit.com

Link to GitHub: https://github.com/divamgupta/stable-diffusion-tensorflow

[Divam Gupta](https://twitter.com/divamgupta/status/1571234504320208897) ported Stable Diffusion over to TF/Keras:

- Converted pre-trained models
- Easy to understand code
- Minimal code footprint

He also released a [Colab](https://colab.research.google.com/drive/1fBlfPsL5DEscub0O_3oJNnJbON4w3EKq?usp=sharing) with Gradio demo.

Should be easy to add TPU / multi-GPU support for inference via Keras. Would be interesting to see if the Keras model can be used on TFlite on embedded / edge devices, something that is difficult to do in the PyTorch version.

diffusion keras machinelearning stable diffusion tensorflow

Data Architect

@ University of Texas at Austin | Austin, TX

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

Staff Software Engineer, Generative AI, Google Cloud AI

@ Google | Mountain View, CA, USA; Sunnyvale, CA, USA

Expert Data Sciences

@ Gainwell Technologies | Any city, CO, US, 99999