add warning
Browse files
model.py
CHANGED
|
@@ -765,6 +765,7 @@ class DatasetConditionedBiencoder(transformers.PreTrainedModel, ContextualModelM
|
|
| 765 |
dataset_backbone: transformers.PreTrainedModel,
|
| 766 |
):
|
| 767 |
super().__init__(config=config)
|
|
|
|
| 768 |
self.backbone = dataset_backbone
|
| 769 |
self.hidden_size = self.backbone.config.hidden_size
|
| 770 |
self.hidden_size = dataset_backbone.config.hidden_size
|
|
|
|
| 765 |
dataset_backbone: transformers.PreTrainedModel,
|
| 766 |
):
|
| 767 |
super().__init__(config=config)
|
| 768 |
+
print("Warning: cde-small-v1 is deprecated as of January 13th, 2025 and may not work correctly with later versions of FlashAttention. Please migrate to cde-small-v2.")
|
| 769 |
self.backbone = dataset_backbone
|
| 770 |
self.hidden_size = self.backbone.config.hidden_size
|
| 771 |
self.hidden_size = dataset_backbone.config.hidden_size
|