Spaces:
Build error
Build error
Update constants.py
Browse files- constants.py +2 -2
constants.py
CHANGED
|
@@ -61,13 +61,13 @@ DOCUMENT_MAP = {
|
|
| 61 |
}
|
| 62 |
|
| 63 |
# Default Instructor Model
|
| 64 |
-
|
| 65 |
|
| 66 |
####
|
| 67 |
#### OTHER EMBEDDING MODEL OPTIONS
|
| 68 |
####
|
| 69 |
|
| 70 |
-
EMBEDDING_MODEL_NAME = "hkunlp/instructor-xl" # Uses 5 GB of VRAM (Most Accurate of all models)
|
| 71 |
# EMBEDDING_MODEL_NAME = "intfloat/e5-large-v2" # Uses 1.5 GB of VRAM (A little less accurate than instructor-large)
|
| 72 |
# EMBEDDING_MODEL_NAME = "intfloat/e5-base-v2" # Uses 0.5 GB of VRAM (A good model for lower VRAM GPUs)
|
| 73 |
# EMBEDDING_MODEL_NAME = "all-MiniLM-L6-v2" # Uses 0.2 GB of VRAM (Less accurate but fastest - only requires 150mb of vram)
|
|
|
|
| 61 |
}
|
| 62 |
|
| 63 |
# Default Instructor Model
|
| 64 |
+
EMBEDDING_MODEL_NAME = "hkunlp/instructor-large" # Uses 1.5 GB of VRAM (High Accuracy with lower VRAM usage)
|
| 65 |
|
| 66 |
####
|
| 67 |
#### OTHER EMBEDDING MODEL OPTIONS
|
| 68 |
####
|
| 69 |
|
| 70 |
+
# EMBEDDING_MODEL_NAME = "hkunlp/instructor-xl" # Uses 5 GB of VRAM (Most Accurate of all models)
|
| 71 |
# EMBEDDING_MODEL_NAME = "intfloat/e5-large-v2" # Uses 1.5 GB of VRAM (A little less accurate than instructor-large)
|
| 72 |
# EMBEDDING_MODEL_NAME = "intfloat/e5-base-v2" # Uses 0.5 GB of VRAM (A good model for lower VRAM GPUs)
|
| 73 |
# EMBEDDING_MODEL_NAME = "all-MiniLM-L6-v2" # Uses 0.2 GB of VRAM (Less accurate but fastest - only requires 150mb of vram)
|