Fix short-conv padding masking on transformers >=5

README.md

@@ -26,7 +26,6 @@ library_name: PyLate
 license: other
 license_name: lfm1.0
 license_link: LICENSE
-base_model: LiquidAI/LFM2.5-350M-Base
 ---
 
 <center>
@@ -55,10 +54,7 @@ Both models are 350M params and the first bidirectional members of the LFM famil
 
 Find more details about the bidirectional architecture and training recipe in our [blog post](https://www.liquid.ai/blog/lfm2-5-retrievers).
 
-> [!NOTE]
-> 💻 **Demo**: https://huggingface.co/spaces/LiquidAI/colbert-tool-selection
-
-![colb](https://cdn-uploads.huggingface.co/production/uploads/63f389fda096536aeaae0a66/snq-O6VmWRJNMYCNHGtDD.png)
+![colbert_retrieval](https://cdn-uploads.huggingface.co/production/uploads/63f389fda096536aeaae0a66/8J_3tsS-wDq77qeDpDgOd.png)
 
 ## 📄 Model details
 
@@ -68,11 +64,13 @@ Find more details about the bidirectional architecture and training recipe in ou
 | **Total parameters**  | ~353M                                  | ~354M                               |
 | **Backbone**          | [LFM2.5-350M-Base](https://huggingface.co/LiquidAI/LFM2.5-350M-Base) + bi-directional patches  |  [LFM2.5-350M-Base](https://huggingface.co/LiquidAI/LFM2.5-350M-Base) + bi-directional patches |
 | **Layers**            | 17 (10 conv + 6 attn + 1 dense)        | 17 (10 conv + 6 attn + 1 pool)      |
+| **Context length**    | 32,768 tokens                          | 32,768 tokens                       |
 | **Vocabulary size**   | 64,402                                 | 65,536                              |
 | **Output**            | 128-dim per token                      | 1024-dim CLS vector                 |
 | **Similarity**        | MaxSim                                 | Cosine                              |
 | **Training precision**| BF16                                   | BF16                                |
 | **License**           | LFM Open License v1.0                  | LFM Open License v1.0               |
+| **Architecture**      | <img src="https://cdn-uploads.huggingface.co/production/uploads/63f389fda096536aeaae0a66/neXdblhfapp1Oln8M9J4V.png" width="250">| <img src="https://cdn-uploads.huggingface.co/production/uploads/63f389fda096536aeaae0a66/y4Lx5eS3hZCRv1g3r1e9i.png" width="250"> |
 
 **Document length:** 512 tokens &nbsp;&nbsp;