Views
No views yet
pip install model2vec[inference]from_pretrained method:1from model2vec.inference import StaticModelPipeline
2
3# Load a pretrained Model2Vec model
4model = StaticModelPipeline.from_pretrained("model_ca_m2v-256-roberta-base-ca-v2-cased-te")
5
6# Predict labels
7predicted = model.predict(["Example sentence"])@article{minishlab2024model2vec,
author = {Tulkens, Stephan and {van Dongen}, Thomas},
title = {Model2Vec: Fast State-of-the-Art Static Embeddings},
year = {2024},
url = {https://github.com/MinishLab/model2vec}
}