wylum commited on
Commit
449b7ee
·
verified ·
1 Parent(s): 3082cf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -263,8 +263,8 @@ def build_qa_chain(collection_name, vector_db, file: str):
263
 
264
  """
265
  #vincent for new LLM
266
- #llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
267
- llm_model = "meta-llama/Llama-2-7b-chat-hf"
268
  llm = HuggingFaceEndpoint(
269
  repo_id=llm_model,
270
  task="text-generation", # Explicitly specify task
 
263
 
264
  """
265
  #vincent for new LLM
266
+ llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
267
+ #llm_model = "meta-llama/Llama-2-7b-chat-hf"
268
  llm = HuggingFaceEndpoint(
269
  repo_id=llm_model,
270
  task="text-generation", # Explicitly specify task