Add instructor model

#7
by awinml - opened
Files changed (1) hide show
  1. utils/models.py +1 -1
utils/models.py CHANGED
@@ -99,7 +99,7 @@ def get_sgpt_embedding_model():
99
  @st.experimental_singleton
100
  def get_instructor_embedding_model():
101
  device = "cuda" if torch.cuda.is_available() else "cpu"
102
- model = INSTRUCTOR("hkunlp/instructor-large")
103
  return model
104
 
105
 
 
99
  @st.experimental_singleton
100
  def get_instructor_embedding_model():
101
  device = "cuda" if torch.cuda.is_available() else "cpu"
102
+ model = INSTRUCTOR("hkunlp/instructor-xl")
103
  return model
104
 
105