Update app.py
Browse files
app.py
CHANGED
@@ -127,7 +127,7 @@ def initialize_llmchain(temperature, max_tokens, top_k, vector_db, progress=gr.P
|
|
127 |
llm = HuggingFaceHub(repo_id=llm_model, model_kwargs={"temperature": temperature,
|
128 |
"max_new_tokens": max_tokens,
|
129 |
"top_k": top_k,
|
130 |
-
"load_in_8bit":
|
131 |
|
132 |
progress(0.75, desc="Defining buffer memory...")
|
133 |
#memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
|
|
|
127 |
llm = HuggingFaceHub(repo_id=llm_model, model_kwargs={"temperature": temperature,
|
128 |
"max_new_tokens": max_tokens,
|
129 |
"top_k": top_k,
|
130 |
+
"load_in_8bit": True})
|
131 |
|
132 |
progress(0.75, desc="Defining buffer memory...")
|
133 |
#memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
|