Spaces:
Runtime error
Runtime error
ffreemt
commited on
Commit
·
fdb2623
1
Parent(s):
055462a
Update paraphrase-multilingual-MiniLM-L12-v2 about 3 times faster
Browse files
app.py
CHANGED
@@ -132,8 +132,15 @@ CHROMA_SETTINGS = Settings(
|
|
132 |
|
133 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
134 |
|
135 |
-
|
|
|
|
|
|
|
|
|
|
|
136 |
# 'max_seq_length': 128
|
|
|
|
|
137 |
|
138 |
# opanai max 4097
|
139 |
# retriever default k = 4, query lenght about CHUNK_SIZE
|
|
|
132 |
|
133 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
134 |
|
135 |
+
# https://jonathansoma.com/words/multi-language-qa-gpt.html
|
136 |
+
# from langchain.embeddings import HuggingFaceEmbeddings
|
137 |
+
# embeddings = HuggingFaceEmbeddings(model_name='paraphrase-multilingual-MiniLM-L12-v2')
|
138 |
+
# https://www.sbert.net/docs/pretrained_models.html
|
139 |
+
#
|
140 |
+
|
141 |
# 'max_seq_length': 128
|
142 |
+
MODEL_NAME = "paraphrase-multilingual-mpnet-base-v2" # 1.11G
|
143 |
+
MODEL_NAME = "paraphrase-multilingual-MiniLM-L12-v2" # 471M
|
144 |
|
145 |
# opanai max 4097
|
146 |
# retriever default k = 4, query lenght about CHUNK_SIZE
|