Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|