fix post coarse transformer gateloop layername

lucidrains · lucidrains · commit 36d8dbac9f16 · 2024-06-20T12:02:29.000-07:00
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'

Original file line number	Diff line number	Diff line change
`@@ -1603,7 +1603,7 @@ def forward_on_codes(`
`1603`	`1603`	`)`
`1604`	`1604`
`1605`	`1605`	`if exists(self.coarse_post_gateloop_block):`
`1606`		`- face_codes, coarse_post_gateloop_cache = self.coarse_gateloop_block(face_codes, cache = coarse_post_gateloop_cache)`
	`1606`	`+ face_codes, coarse_post_gateloop_cache = self.coarse_post_gateloop_block(face_codes, cache = coarse_post_gateloop_cache)`
`1607`	`1607`
`1608`	`1608`	`else:`
`1609`	`1609`	`attended_face_codes = None`