vishwask commited on
Commit
e780d85
·
verified ·
1 Parent(s): 2ac0a44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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": False})
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)