diff --git a/setup.cfg b/setup.cfg
index 191b6cd..1ac758f 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = v-diffusion-pytorch
-version = 0.0.1
+version = 0.0.2
 author = Katherine Crowson
 author_email = crowsonkb@gmail.com
 url = https://github.com/crowsonkb/v-diffusion-pytorch