deepyuna commited on
Commit
49de155
·
verified ·
1 Parent(s): 22c204f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -10,6 +10,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
10
  from peft import PeftModel
11
  import shutil
12
 
 
 
 
 
13
 
14
  HF_TOKEN_LLAMA = os.environ.get("HF_TOKEN_LLAMA")
15
  login(token=HF_TOKEN_LLAMA)
 
10
  from peft import PeftModel
11
  import shutil
12
 
13
+ #___________ cache issue 03_08
14
+ if os.path.exists("private_space_cache"):
15
+ shutil.rmtree("private_space_cache") # Forcefully remove corrupted cache folder
16
+ #____________________________________
17
 
18
  HF_TOKEN_LLAMA = os.environ.get("HF_TOKEN_LLAMA")
19
  login(token=HF_TOKEN_LLAMA)