Skip to content

Instantly share code, notes, and snippets.

@horoiwa

horoiwa/diffusion.py

Created July 11, 2023 09:00
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save horoiwa/550da5c3d57efefac43a49a02b7d4a37 to your computer and use it in GitHub Desktop.
Save horoiwa/550da5c3d57efefac43a49a02b7d4a37 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
diffusion.py
eps = tf.random.normal(shape=x_0.shape, mean=0., stddev=1.)
x_t = tf.sqrt(alphas_cumprod_t) * x_0 + tf.sqrt(1. - alphas_cumprod_t) * eps
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment