Web: https://www.reddit.com/r/MachineLearning/comments/xhi9ci/p_stable_diffusion_in_tensorflow_keras/

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

Machine Learning 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

Research Scientists

@ ODU Research Foundation | Norfolk, Virginia

Embedded Systems Engineer (Robotics)

@ Neo Cybernetica | Bedford, New Hampshire

2023 Luis J. Alvarez and Admiral Grace M. Hopper Postdoc Fellowship in Computing Sciences

@ Lawrence Berkeley National Lab | San Francisco, CA

Senior Manager Data Scientist

@ NAV | Remote, US

Senior AI Research Scientist

@ Earth Species Project | Remote anywhere

Research Fellow- Center for Security and Emerging Technology (Multiple Opportunities)

@ University of California Davis | Washington, DC

Staff Fellow - Data Scientist

@ U.S. FDA/Center for Devices and Radiological Health | Silver Spring, Maryland

Staff Fellow - Senior Data Engineer

@ U.S. FDA/Center for Devices and Radiological Health | Silver Spring, Maryland

Machine Learning Data Engineer Intern (Jyoti Dharna)

@ Benson Hill | St. Louis, Missouri

Software Engineer / SDE I, Chime SDK Video Research Engineering

@ Amazon.com | East Palo Alto, California, USA

IND (New) Senior ML Ops Engineer - WiQ

@ Quantium | Hyderabad

Data Engineer

@ LendingTree | Remote