Skip to content

set_cache_dir

set_cache_dir

Configuration of HuggingFace and Torch cache directories.

resolve_cache_directory 

resolve_cache_directory()

Return the model cache root from VERSATIL_CACHE_DIR or the default.

Source code in src/versatil/common/set_cache_dir.py 
 9
10
11
def resolve_cache_directory() -> Path:
    """Return the model cache root from ``VERSATIL_CACHE_DIR`` or the default."""
    return Path(os.environ.get("VERSATIL_CACHE_DIR", str(DEFAULT_CACHE_DIR)))

setup_cache_directories 

setup_cache_directories(cache_dir)

Configure cache directories for model downloads before any run.

Source code in src/versatil/common/set_cache_dir.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
def setup_cache_directories(cache_dir: str | Path) -> None:
    """Configure cache directories for model downloads before any run."""
    cache_path = Path(cache_dir)

    os.environ["HF_HOME"] = str(cache_path / "huggingface")
    os.environ["HF_HUB_CACHE"] = str(cache_path / "huggingface" / "hub")
    os.environ["TORCH_HOME"] = str(cache_path / "torch")

    for dir_path in [
        cache_path / "huggingface" / "transformers",
        cache_path / "huggingface" / "hub",
        cache_path / "torch" / "hub",
    ]:
        dir_path.mkdir(parents=True, exist_ok=True)