dkdaniz commited on
Commit
bc0667c
1 Parent(s): 8744644

Update constants.py

Browse files
Files changed (1) hide show
  1. constants.py +2 -2
constants.py CHANGED
@@ -61,13 +61,13 @@ DOCUMENT_MAP = {
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)
 
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)