Update.

diff --git a/minidiffusion.py b/minidiffusion.py
index e7be8c1..066cbbb 100755 (executable)
--- a/minidiffusion.py
+++ b/minidiffusion.py
@@ -66,12 +66,14 @@ def sample_mnist(nb):
     return result
 
 samplers = {
-    'gaussian_mixture': sample_gaussian_mixture,
-    'ramp': sample_ramp,
-    'two_discs': sample_two_discs,
-    'disc_grid': sample_disc_grid,
-    'spiral': sample_spiral,
-    'mnist': sample_mnist,
+    f.__name__.removeprefix('sample_') : f for f in [
+        sample_gaussian_mixture,
+        sample_ramp,
+        sample_two_discs,
+        sample_disc_grid,
+        sample_spiral,
+        sample_mnist,
+    ]
 }
 
 ######################################################################

diff --git a/minidiffusion_gaussian_mixture.pdf b/minidiffusion_gaussian_mixture.pdf
index 04d9b3d..4960e3e 100644 (file)
Binary files a/minidiffusion_gaussian_mixture.pdf and b/minidiffusion_gaussian_mixture.pdf differ

diff --git a/minidiffusion_ramp.pdf b/minidiffusion_ramp.pdf
index 17c11b8..8fc2c68 100644 (file)
Binary files a/minidiffusion_ramp.pdf and b/minidiffusion_ramp.pdf differ