No official torrent yet — requesting it tells the flock what to host next. You cannot seed a catalog-only page.
Cataloged from sentence-transformers/allenai-specter
All models by sentence-transformers →This model is a conversion of the AllenAI SPECTER model to sentence-transformers. It can be used to map the titles & abstracts of scientific publications to a vector space such that similar papers are close.
Using this model becomes easy when you have sentence-transformers installed:
Then you can use the model like this:
Without sentence-transformers, you can use the model like this: First, you pass your input through the transformer model, then you have to apply the right pooling-operation on-top of the contextualized word embeddings.
See AllenAI SPECTER
pip install -U sentence-transformersfrom sentence_transformers import SentenceTransformersentences = ["This is an example sentence", "Each sentence is converted"]
model = SentenceTransformer('sentence-transformers/allenai-specter')embeddings = model.encode(sentences)print(embeddings)from transformers import AutoTokenizer, AutoModelimport torch
def cls_pooling(model_output, attention_mask): return model_output[0][:,0]
# Sentences we want sentence embeddings forsentences = ['This is an example sentence', 'Each sentence is converted']
# Load model from HuggingFace Hubtokenizer = AutoTokenizer.from_pretrained('sentence-transformers/allenai-specter')model = AutoModel.from_pretrained('sentence-transformers/allenai-specter')
# Tokenize sentencesencoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
# Compute token embeddingswith torch.no_grad(): model_output = model(**encoded_input)
# Perform pooling. In this case, max pooling.sentence_embeddings = cls_pooling(model_output, encoded_input['attention_mask'])
print("Sentence embeddings:")print(sentence_embeddings)SentenceTransformer( (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: BertModel (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': True, 'pooling_mode_mean_tokens': False, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False}))