NovaSearch
/

stella_en_400M_v5

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

sigridjineth commited on 22 days ago

Commit

e9e84da

·

verified ·

1 Parent(s): db4ace1

Update modeling.py

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -446,7 +446,7 @@ class NewAttention(nn.Module):
         if use_memory_efficient_attention is None:
             use_memory_efficient_attention = self.config.use_memory_efficient_attention
         self.use_memory_efficient_attention = use_memory_efficient_attention
-        self.memory_efficient_attention = None if xops is None else xops.memory_efficient_attention
         if self.use_memory_efficient_attention:
             assert self.memory_efficient_attention is not None, 'please install xformers'
         if self.config.unpad_inputs:

         if use_memory_efficient_attention is None:
             use_memory_efficient_attention = self.config.use_memory_efficient_attention
         self.use_memory_efficient_attention = use_memory_efficient_attention
+        self.memory_efficient_attention = None
         if self.use_memory_efficient_attention:
             assert self.memory_efficient_attention is not None, 'please install xformers'
         if self.config.unpad_inputs: