Spaces:
Build error
Build error
Add instructor model
#7
by
awinml
- opened
- 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-
|
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 |
|