Spaces:
Sleeping
Sleeping
mwitiderrick
commited on
Commit
•
a9ee327
1
Parent(s):
625efa6
Update app.py
Browse files
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/
|
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)
|
|
|
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)
|