yonkoyonks commited on
Commit
427189e
·
verified ·
1 Parent(s): c23a150

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -56,6 +56,7 @@ def query_agent(df: pd.DataFrame, query: str) -> str:
56
  max_new_tokens=1024,
57
  n_ctx=16384,
58
  verbose=True,
 
59
  )
60
 
61
  answer = llm(prompt)
 
56
  max_new_tokens=1024,
57
  n_ctx=16384,
58
  verbose=True,
59
+ n_threads=4,
60
  )
61
 
62
  answer = llm(prompt)