File size: 467 Bytes
1df3d2a
4c655d5
6861e2a
 
1df3d2a
 
4c655d5
 
6861e2a
db81bb8
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from langchain_huggingface import HuggingFaceEmbeddings
from app.models.AzureOpenAI import azure_llm
from app.models.OpenAI import llm as openai_llm
from app.config import model

# model = TinyLlamaModel()
# LLM = 

LLM = openai_llm if model == "OpenAI" else azure_llm

Embedding_model_en = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2") # only English
Embedding_model_de = HuggingFaceEmbeddings(model_name="danielheinz/e5-base-sts-en-de")