Spaces:
Sleeping
Sleeping
Update chain_setup.py
Browse files- chain_setup.py +3 -2
chain_setup.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from huggingface_hub import hf_hub_download
|
2 |
from langchain.llms import LlamaCpp
|
3 |
from langchain.chains import ConversationalRetrievalChain
|
@@ -53,8 +54,8 @@ def load_llm():
|
|
53 |
f16_kv=True,
|
54 |
n_parts=-1,
|
55 |
seed=-1,
|
56 |
-
verbose=True
|
57 |
-
|
58 |
)
|
59 |
|
60 |
return llm
|
|
|
1 |
+
import os
|
2 |
from huggingface_hub import hf_hub_download
|
3 |
from langchain.llms import LlamaCpp
|
4 |
from langchain.chains import ConversationalRetrievalChain
|
|
|
54 |
f16_kv=True,
|
55 |
n_parts=-1,
|
56 |
seed=-1,
|
57 |
+
verbose=True,
|
58 |
+
client=None
|
59 |
)
|
60 |
|
61 |
return llm
|