mirror of
https://github.com/katanemo/plano.git
synced 2026-04-26 09:16:24 +02:00
11 lines
298 B
Python
11 lines
298 B
Python
|
import os
|
||
|
|
import sentence_transformers
|
||
|
|
|
||
|
|
def load_transformers(models = os.getenv("MODELS", "sentence-transformers/all-MiniLM-L6-v2")):
|
||
|
|
transformers = {}
|
||
|
|
|
||
|
|
for model in models.split(','):
|
||
|
|
transformers[model] = sentence_transformers.SentenceTransformer(model)
|
||
|
|
|
||
|
|
return transformers
|