aitextgen icon indicating copy to clipboard operation
aitextgen copied to clipboard

Get embeddings from model

Open minimaxir opened this issue 4 years ago • 0 comments

Two distinct use cases:

  1. Get embeddings corresponding to a specific token (e.g. the embeddings corresponding to an <|endoftext|> token.)
  2. Get embeddings for a full input text word2vec style.

Technically somewhat outside the scope for aitextgen but it's not in base Transformers and I can make clever use of it.

minimaxir avatar May 15 '21 16:05 minimaxir