Skip to content

Commit 5d729f0

Browse files
authored
Update README.md
1 parent 89d0d06 commit 5d729f0

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

README.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,7 @@ model = XTransformer(
2727
dec_num_tokens = 256,
2828
dec_depth = 6,
2929
dec_heads = 8,
30-
dec_max_seq_len = 1024,
31-
enc_num_memory_tokens = 0
30+
dec_max_seq_len = 1024
3231
)
3332

3433
src = torch.randint(0, 256, (1, 1024))

0 commit comments

Comments
 (0)