mwitiderrick commited on
Commit
a9ee327
1 Parent(s): 625efa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
8
  import chainlit as cl
9
  import os
10
 
11
- MODEL_PATH = "hf:neuralmagic/TinyLlama-1.1B-Chat-v0.4-pruned50-quant-ds"
12
- generation_config = {"max_new_tokens":200, "repetition_penalty":1.1, "do_sample":True}
13
  llm = DeepSparse(model=MODEL_PATH, generation_config=generation_config)
14
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={'device': 'cpu'})
15
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=600, chunk_overlap=100)
 
8
  import chainlit as cl
9
  import os
10
 
11
+ MODEL_PATH = "hf:neuralmagic/mpt-7b-chat-pruned50-quant"
12
+ generation_config = {"max_new_tokens":200}
13
  llm = DeepSparse(model=MODEL_PATH, generation_config=generation_config)
14
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={'device': 'cpu'})
15
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=600, chunk_overlap=100)