edugp commited on
Commit
ee732fe
·
1 Parent(s): d131aa3

Add distiluse-base-multilingual-cased-v2

Browse files
Files changed (1) hide show
  1. perplexity_lenses/engine.py +1 -1
perplexity_lenses/engine.py CHANGED
@@ -12,7 +12,7 @@ from sentence_transformers import SentenceTransformer
12
  from perplexity_lenses.visualization import draw_interactive_scatter_plot
13
 
14
  logger = logging.getLogger(__name__)
15
- EMBEDDING_MODELS = ["distiluse-base-multilingual-cased-v1", "all-mpnet-base-v2", "flax-sentence-embeddings/all_datasets_v3_mpnet-base"]
16
  DIMENSIONALITY_REDUCTION_ALGORITHMS = ["UMAP", "t-SNE"]
17
  DOCUMENT_TYPES = ["Whole document", "Sentence"]
18
  SEED = 0
 
12
  from perplexity_lenses.visualization import draw_interactive_scatter_plot
13
 
14
  logger = logging.getLogger(__name__)
15
+ EMBEDDING_MODELS = ["distiluse-base-multilingual-cased-v1", "distiluse-base-multilingual-cased-v2", "all-mpnet-base-v2", "flax-sentence-embeddings/all_datasets_v3_mpnet-base"]
16
  DIMENSIONALITY_REDUCTION_ALGORITHMS = ["UMAP", "t-SNE"]
17
  DOCUMENT_TYPES = ["Whole document", "Sentence"]
18
  SEED = 0