Spaces:
Sleeping
Sleeping
Commit ·
8e21db2
1
Parent(s): 8b31749
fix: update dependencies and use L12-v2 model (better quality, 16GB RAM is enough)
Browse files- Dockerfile +2 -0
- README.md +2 -1
- requirements.txt +5 -2
Dockerfile
CHANGED
|
@@ -18,6 +18,8 @@ ENV PYTHONUNBUFFERED=1
|
|
| 18 |
ENV TRANSFORMERS_CACHE=/home/user/.cache/transformers
|
| 19 |
ENV SENTENCE_TRANSFORMERS_HOME=/home/user/.cache/sentence_transformers
|
| 20 |
ENV HF_HOME=/home/user/.cache/huggingface
|
|
|
|
|
|
|
| 21 |
|
| 22 |
# Copy requirements and install dependencies
|
| 23 |
COPY --chown=user requirements.txt .
|
|
|
|
| 18 |
ENV TRANSFORMERS_CACHE=/home/user/.cache/transformers
|
| 19 |
ENV SENTENCE_TRANSFORMERS_HOME=/home/user/.cache/sentence_transformers
|
| 20 |
ENV HF_HOME=/home/user/.cache/huggingface
|
| 21 |
+
ENV EMBEDDING_MODEL=sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2
|
| 22 |
+
ENV EMBEDDING_DIMENSIONS=384
|
| 23 |
|
| 24 |
# Copy requirements and install dependencies
|
| 25 |
COPY --chown=user requirements.txt .
|
README.md
CHANGED
|
@@ -29,9 +29,10 @@ app_port: 7860
|
|
| 29 |
|
| 30 |
## Модель
|
| 31 |
|
| 32 |
-
Используется модель: `paraphrase-multilingual-MiniLM-
|
| 33 |
- Поддержка 50+ языков (включая русский)
|
| 34 |
- Размерность векторов: 384
|
|
|
|
| 35 |
- Оптимизирована для семантического поиска
|
| 36 |
|
| 37 |
## Endpoints
|
|
|
|
| 29 |
|
| 30 |
## Модель
|
| 31 |
|
| 32 |
+
Используется модель: `paraphrase-multilingual-MiniLM-L12-v2`
|
| 33 |
- Поддержка 50+ языков (включая русский)
|
| 34 |
- Размерность векторов: 384
|
| 35 |
+
- 12 слоёв (лучшее качество чем L6-v2)
|
| 36 |
- Оптимизирована для семантического поиска
|
| 37 |
|
| 38 |
## Endpoints
|
requirements.txt
CHANGED
|
@@ -3,12 +3,15 @@
|
|
| 3 |
|
| 4 |
fastapi==0.104.1
|
| 5 |
uvicorn[standard]==0.24.0
|
| 6 |
-
sentence-transformers==2.2.2
|
| 7 |
numpy>=1.24.0,<2.0.0
|
| 8 |
pydantic==2.5.3
|
| 9 |
python-dotenv==1.0.0
|
| 10 |
|
| 11 |
# PyTorch - используем стандартную версию (HuggingFace имеет достаточно памяти)
|
| 12 |
torch>=2.1.0,<2.2.0
|
| 13 |
-
transformers
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
|
|
|
|
| 3 |
|
| 4 |
fastapi==0.104.1
|
| 5 |
uvicorn[standard]==0.24.0
|
|
|
|
| 6 |
numpy>=1.24.0,<2.0.0
|
| 7 |
pydantic==2.5.3
|
| 8 |
python-dotenv==1.0.0
|
| 9 |
|
| 10 |
# PyTorch - используем стандартную версию (HuggingFace имеет достаточно памяти)
|
| 11 |
torch>=2.1.0,<2.2.0
|
| 12 |
+
transformers==4.36.2
|
| 13 |
+
|
| 14 |
+
# Sentence Transformers с совместимой версией huggingface_hub
|
| 15 |
+
sentence-transformers==2.3.1
|
| 16 |
+
huggingface_hub>=0.19.0,<0.20.0
|
| 17 |
|