From: Francois Fleuret <francois@fleuret.org>
Date: Thu, 18 Aug 2022 06:15:28 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=87d376ff7929347865b199d05d003ab3b168f249;p=pytorch.git

Update.
---

diff --git a/minidiffusion.py b/minidiffusion.py
index e7be8c1..066cbbb 100755
--- 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
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
Binary files a/minidiffusion_ramp.pdf and b/minidiffusion_ramp.pdf differ