Spaces:
Build error
Build error
Update utils.py
Browse files
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)
|