jxm
/

cde-small-v1

Feature Extraction

sentence-transformers

Eval Results (legacy)

Model card Files Files and versions

jxm commited on Jan 21, 2025

Commit

94384fc

·

1 Parent(s): 8d57361

add warning

Files changed (1) hide show

model.py +1 -0

model.py CHANGED Viewed

@@ -765,6 +765,7 @@ class DatasetConditionedBiencoder(transformers.PreTrainedModel, ContextualModelM
             dataset_backbone: transformers.PreTrainedModel,
         ):
         super().__init__(config=config)
         self.backbone = dataset_backbone
         self.hidden_size = self.backbone.config.hidden_size
         self.hidden_size = dataset_backbone.config.hidden_size

             dataset_backbone: transformers.PreTrainedModel,
         ):
         super().__init__(config=config)
+        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.")
         self.backbone = dataset_backbone
         self.hidden_size = self.backbone.config.hidden_size
         self.hidden_size = dataset_backbone.config.hidden_size