diff --git a/agent/model_metadata.py b/agent/model_metadata.py
index 335d02bd6f..11a998862b 100644
--- a/agent/model_metadata.py
+++ b/agent/model_metadata.py
@@ -203,6 +203,7 @@ _CONTEXT_LENGTH_KEYS = (
     "max_seq_len",
     "n_ctx_train",
     "n_ctx",
+    "ctx_size",
 )
 
 _MAX_COMPLETION_KEYS = (