Spaces:
Paused
Paused
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)
|