File size: 677 Bytes
1489a02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
import shutil
def free_up_space():
# Check disk usage before cleanup
print("Disk usage before cleanup:")
os.system("df -h")
# Clear Hugging Face cache
print("Clearing Hugging Face cache...")
shutil.rmtree(os.path.expanduser("~/.cache/huggingface"), ignore_errors=True)
shutil.rmtree(os.path.expanduser("~/.cache/transformers"), ignore_errors=True)
# Clear PyTorch cache
print("Clearing PyTorch cache...")
shutil.rmtree(os.path.expanduser("~/.cache/torch"), ignore_errors=True)
# Check disk usage after cleanup
print("Disk usage after cleanup:")
os.system("df -h")
if __name__ == "__main__":
free_up_space() |