Update.

diff --git a/minidiffusion.py b/minidiffusion.py
index cbdb142..ad1cda0 100755 (executable)
--- a/minidiffusion.py
+++ b/minidiffusion.py
@@ -81,6 +81,8 @@ ax.hist(d, 25, (-1, 1), histtype = 'stepfilled', color = 'lightblue', density =
 d = x.flatten().detach().numpy()
 ax.hist(d, 25, (-1, 1), histtype = 'step', color = 'red', density = True, label = 'Synthesis')
 
+ax.legend(frameon = False, loc = 2)
+
 filename = 'diffusion.pdf'
 fig.savefig(filename, bbox_inches='tight')