[CLEANUP]

kyegomez · Sep 10, 2024 · d14c1b4 · d14c1b4
1 parent 6e0773b
commit d14c1b4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/chai_one/model.py b/chai_one/model.py
@@ -88,7 +88,7 @@ def forward(
                 pairwise_repr=msa_feats,
                 attn_bias=attn_bias,
             )
-            
+
         logger.info(f"MSA feats shape: {msa_feats.shape}")
 
         return msa_feats
diff --git a/example.py b/example.py
@@ -9,7 +9,7 @@
 dim_msa = 128
 dim_msa_input = 134  # Adjusted to match the expected input dimension
 dim_additional_msa_feats = 2
-window_size = None
+window_size = 25
 
 # Initialize the model
 logger.info("Initializing ChaiOne model")