JohnSnowLabs / langtest

Deliver safe & effective language models
http://langtest.org/
Apache License 2.0
490 stars 37 forks source link

Feature/run embedding benchmark pipelines CLI #960

Closed ArshaanNazir closed 7 months ago

ArshaanNazir commented 7 months ago

Description

Running embedding benchmarking with CLI

Sets Keys:

python -m langtest config set OPENAI_API_KEY=<KEY>

python -m langtest config set HUGGINGFACEHUB_API_TOKEN=<HF_TOKEN>

Example Usage (Single Model):

python -m langtest benchmark embeddings --model TaylorAI/bge-micro --hub huggingface

image

Example Usage (Multiple Models):

python -m langtest benchmark embeddings --model "TaylorAI/bge-micro,TaylorAI/gte-tiny,intfloat/e5-small" --hub huggingface

image