Add Support for mistralai/Mistral-Small-3.1-24B-Instruct-2503 model

Author: nikileshx

models/tt_transformers/tt/model_config.py

@@ -1380,6 +1380,8 @@ def prepare_residual_tensor_prefill(self, x_bsh, force_replicated=False):
 
     def _get_text_prefix(self):
         if self.is_vision():
+            if "Mistral-Small-3.1-24B-Instruct-2503" in self.model_name:
+                return "language_model."
             return "text_model."
         else:
             return ""