isayahc commited on
Commit
8bddd83
·
1 Parent(s): 1c2a1d9

adding some llm configurations

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -18,6 +18,10 @@ from langchain.document_loaders import WebBaseLoader
18
 
19
  from typing import Iterator
20
 
 
 
 
 
21
 
22
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=350, chunk_overlap=10)
23
  model_id = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})
 
18
 
19
  from typing import Iterator
20
 
21
+ MAX_MAX_NEW_TOKENS = 2048
22
+ DEFAULT_MAX_NEW_TOKENS = 1024
23
+ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
24
+
25
 
26
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=350, chunk_overlap=10)
27
  model_id = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})