Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -23,10 +23,9 @@ def define_embeddings_llm(openai_key):
|
|
23 |
embeddings = HuggingFaceEmbeddings(
|
24 |
model_name="sentence-transformers/all-MiniLM-L6-v2"
|
25 |
)
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
)
|
30 |
langchain.llm_cache = InMemoryCache()
|
31 |
return embeddings,llm
|
32 |
|
|
|
23 |
embeddings = HuggingFaceEmbeddings(
|
24 |
model_name="sentence-transformers/all-MiniLM-L6-v2"
|
25 |
)
|
26 |
+
model_id = "lmsys/vicuna-7b-v1.3"
|
27 |
+
pipe = pipeline("text-generation", model=model, max_length=2048)
|
28 |
+
llm = HuggingFacePipeline(pipeline=pipe)
|
|
|
29 |
langchain.llm_cache = InMemoryCache()
|
30 |
return embeddings,llm
|
31 |
|