Update modeling_reasonir_8b.py
Browse files- modeling_reasonir_8b.py +1 -2
modeling_reasonir_8b.py
CHANGED
|
@@ -1670,8 +1670,7 @@ class LlamaForTokenClassification(LlamaPreTrainedModel):
|
|
| 1670 |
attentions=outputs.attentions,
|
| 1671 |
)
|
| 1672 |
|
| 1673 |
-
|
| 1674 |
-
class ReasonIRModel(LLamaModel):
|
| 1675 |
"""
|
| 1676 |
ReasonIRModel is a wrapper around LlamaModel with bi-directional attention for retrieval tasks
|
| 1677 |
"""
|
|
|
|
| 1670 |
attentions=outputs.attentions,
|
| 1671 |
)
|
| 1672 |
|
| 1673 |
+
class ReasonIRModel(LlamaModel):
|
|
|
|
| 1674 |
"""
|
| 1675 |
ReasonIRModel is a wrapper around LlamaModel with bi-directional attention for retrieval tasks
|
| 1676 |
"""
|