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("atc-pilot-speaker-role-potion-base-32M")
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}
}