How to run Huemin's JAX Diffusion on RunPod

BY ZHEN LU - AUGUST 9, 2022
Here's a quick and dirty guide to getting Huemin's JAX diffusion to run on runpod!

Today I figured I'd switch things up a bit and leave you a gif quick start instead of the usual article. It only takes a few minutes to get started with this and all the deps have been sorted out for your in the nb. If you need the notebook file that you should upload to your instance, you can download it from one of these two places:

https://github.com/Run-Pod/notebooks/blob/main/Huemin_Jax_Diffusion_2_7-RunPod.ipynb

https://colab.research.google.com/github/Run-Pod/notebooks/blob/main/Huemin_Jax_Diffusion_2_7-RunPod.ipynb

Please note that it will sit here for quite a few minutes while it downloads models and prepared for the first render cycle:

Be patient and it will start rendering when it's ready. I was getting around 1.5-2s/it on the demo example.

Enjoy!

Credits to:

https://twitter.com/huemin_art for the notebook

https://twitter.com/AlexanderRedde3 for working out the deps

https://twitter.com/nshepperd1 and https://twitter.com/RiversHaveWings for jax diffusion

Zhen is one of the co-founders of RunPod. He holds a PhD in Computational Chemistry and has lots of experience teaching, coding, and talking to people.

Want to join our growing community?
Chat with us on our Discord Server.
Prefer email? Contact us at support@runpod.io