fix post coarse transformer gateloop layername

lucidrains · Jun 20, 2024 · 36d8dba · 36d8dba
1 parent 9db2e34
commit 36d8dba
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -1603,7 +1603,7 @@ def forward_on_codes(
             )
 
             if exists(self.coarse_post_gateloop_block):
-                face_codes, coarse_post_gateloop_cache = self.coarse_gateloop_block(face_codes, cache = coarse_post_gateloop_cache)
+                face_codes, coarse_post_gateloop_cache = self.coarse_post_gateloop_block(face_codes, cache = coarse_post_gateloop_cache)
 
         else:
             attended_face_codes = None

diff --git a/meshgpt_pytorch/version.py b/meshgpt_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.5.0'
+__version__ = '1.5.1'