From 5ce704ebcbb6533d063b9f5c91446aee543abd88 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Thu, 21 Nov 2024 09:04:07 -0800
Subject: [PATCH] fix an issue with new pytorch + torchtyping

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 0177303..ad44380 100644
--- 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 = "lucidrains@gmail.com" }
@@ -34,6 +34,7 @@ dependencies = [
     'scipy',
     "torch>=2.5",
     'torchdiffeq',
+    'torchtyping>=0.1.5',
     "tqdm"
 ]