Skip to content

Commit 0476b4e

Browse files
committed
fix types
1 parent 925d80a commit 0476b4e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

transfusion_pytorch/transfusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -363,7 +363,7 @@ def derive_rotary_positions_from_modality_positions(
363363
def embed_modality_tokens(
364364
seq_len: int,
365365
dim: int,
366-
modality_tokens: list[list[Float['... d'] | Float['d ...']]],
366+
modality_tokens: list[list[Float['...']]],
367367
modalities: Int['b m 3'],
368368
modality_id: int,
369369
channel_first: bool

0 commit comments

Comments
 (0)