🐛 access via config

ibm-granite · Jan 15, 2025 · 04c925d · 04c925d
1 parent ccde10d
commit 04c925d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tsfm_public/models/tinytimemixer/modeling_tinytimemixer.py b/tsfm_public/models/tinytimemixer/modeling_tinytimemixer.py
@@ -1840,7 +1840,9 @@ def forward(
             )
 
         sequence_length = (
-            self.config.masked_context_length if self.config.masked_context_length is not None else self.context_length
+            self.config.masked_context_length
+            if self.config.masked_context_length is not None
+            else self.config.context_length
         )
 
         if past_values.shape[1] > sequence_length: