Samurai719214 commited on
Commit
0ee3094
·
verified ·
1 Parent(s): 4ed952a

Update cache_manager.py

Browse files
Files changed (1) hide show
  1. cache_manager.py +6 -11
cache_manager.py CHANGED
@@ -1,13 +1,8 @@
1
- import shutil
2
- import os
3
 
4
- cache_dirs = [
5
- os.path.expanduser("~/.cache/huggingface/transformers"),
6
- os.path.expanduser("~/.cache/huggingface/datasets"),
7
- os.path.expanduser("~/.cache/huggingface/hub"),
8
- ]
9
 
10
- for d in cache_dirs:
11
- if os.path.exists(d):
12
- shutil.rmtree(d)
13
- print(f"Deleted cache: {d}")
 
1
+ from huggingface_hub import scan_cache_dir, delete_cache_dir
 
2
 
3
+ # Inspect current cache
4
+ cache_info = scan_cache_dir()
5
+ print(cache_info)
 
 
6
 
7
+ # Delete whole cache (careful: removes ALL models/datasets cached)
8
+ delete_cache_dir()