feat: version bump

Author: AshishKumar4

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "flaxdiff"
-version = "0.2.0"
+version = "0.2.1"
 description = "A versatile and easy to understand Diffusion library"
 readme = "README.md"
 authors = [