Geraldine commited on
Commit
adf7988
·
1 Parent(s): 8a2f928

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def define_embeddings_llm(openai_key):
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
 
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_id, max_length=2048)
28
  llm = HuggingFacePipeline(pipeline=pipe)
29
  langchain.llm_cache = InMemoryCache()
30
  return embeddings,llm