fix an issue with new pytorch + torchtyping

lucidrains · Nov 21, 2024 · 5ce704e · 5ce704e
1 parent c1c93e6
commit 5ce704e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "pi-zero-pytorch"
-version = "0.0.27"
+version = "0.0.28"
 description = "π0 in Pytorch"
 authors = [
     { name = "Phil Wang", email = "[email protected]" }
@@ -34,6 +34,7 @@ dependencies = [
     'scipy',
     "torch>=2.5",
     'torchdiffeq',
+    'torchtyping>=0.1.5',
     "tqdm"
 ]