HemaMeena commited on
Commit
f391e55
·
verified ·
1 Parent(s): 449a65e

Update app.py

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