This commit is contained in:
Gokul Mohanarangan
2023-08-30 10:43:51 +05:30
parent df078f7bd6
commit 012390d0aa
2 changed files with 36 additions and 8 deletions

View File

@@ -38,7 +38,7 @@ def migrate_cache_llm():
from transformers.utils.hub import move_cache
print("Moving LLM cache")
move_cache()
move_cache(cache_dir=IMAGE_MODEL_DIR)
print("LLM cache moved")