Spaces:

awinml
/

2-qa-earnings-sentencewise

Build error

awinml commited on May 21, 2023

Commit

e70a92a

•

1 Parent(s): 4f49100

Add instructor model

Files changed (1) hide show

utils/models.py CHANGED Viewed

@@ -99,7 +99,7 @@ def get_sgpt_embedding_model():
 @st.experimental_singleton
 def get_instructor_embedding_model():
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    model = INSTRUCTOR("hkunlp/instructor-large")
     return model

 @st.experimental_singleton
 def get_instructor_embedding_model():
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    model = INSTRUCTOR("hkunlp/instructor-xl")
     return model