Update README.md

kyegomez · Jan 7, 2024 · 5ce5b8e · 5ce5b8e
1 parent a95867f
commit 5ce5b8e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -37,6 +37,7 @@ model = MultiModalMambaBlock(
     encoder_dim = 64,  # Dimension of the encoder token embeddings
     encoder_depth = 5,  # Number of encoder transformer layers
     encoder_heads = 4  # Number of encoder attention heads
+    fusion_method="mlp",
 )
 
 # Pass the input tensors through the model