Update app.py
Browse files
app.py
CHANGED
@@ -39,6 +39,13 @@ import locale
|
|
39 |
import time
|
40 |
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
class CFG:
|
43 |
# LLMs
|
44 |
model_name = 'llama2-13b-chat' # wizardlm, llama2-7b-chat, llama2-13b-chat, mistral-7B
|
|
|
39 |
import time
|
40 |
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
41 |
|
42 |
+
from transformers import logging
|
43 |
+
import shutil
|
44 |
+
|
45 |
+
# Clear transformers cache
|
46 |
+
logging.set_verbosity_error()
|
47 |
+
shutil.rmtree('./.cache', ignore_errors=True)
|
48 |
+
|
49 |
class CFG:
|
50 |
# LLMs
|
51 |
model_name = 'llama2-13b-chat' # wizardlm, llama2-7b-chat, llama2-13b-chat, mistral-7B
|