credit assignment to Edward Hu, young ivy league student

lucidrains · Mar 8, 2025 · e7c8fc6 · e7c8fc6
1 parent 836af4a
commit e7c8fc6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "x-transformers"
-version = "2.1.14"
+version = "2.1.15"
 description = "X-Transformers"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }

diff --git a/tests/test_x_transformers.py b/tests/test_x_transformers.py
@@ -739,5 +739,5 @@ def test_belief_state_wrapper(
     if goal_suffix:
         suffix = torch.randint(0, 20000, (2, 2))
 
-    sampled = model.generate_with_suffix_token_only(seq[:, :1], 16, suffix = suffix)
+    sampled = model.generate_with_suffix_cond(seq[:, :1], 16, suffix = suffix)
     assert sampled.shape == (2, 16)
diff --git a/x_transformers/belief_state_wrapper.py b/x_transformers/belief_state_wrapper.py
@@ -1,7 +1,7 @@
 
 # Belief State Transformer
 
-# https://arxiv.org/abs/2410.23506
+# Hu et al. https://arxiv.org/abs/2410.23506
 # https://www.youtube.com/watch?v=aqhbRtB2Fyg
 
 from __future__ import annotations
@@ -107,7 +107,7 @@ def __init__(
 
     @torch.no_grad()
     @eval_decorator
-    def generate_with_suffix_token_only(
+    def generate_with_suffix_cond(
         self,
         prompts,
         seq_len,