ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Dedup the HF hub cache-dir path construction into a helper

Browse source
This commit is contained in:
oobabooga 2026-07-04 11:09:37 -03:00
commit 4161d5d87a
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
studio/backend/core/inference/diffusion.py
View file

@ -806,10 +806,15 @@ class DiffusionBackend:
return total, base_files
@staticmethod
def _cache_bytes(repo_id: str) -> int:
def _hub_cache_repo_dir(repo_id: str) -> Path:
"""The local HF hub cache dir for ``repo_id`` (``.../models--org--name``)."""
from huggingface_hub import constants
blobs = Path(constants.HF_HUB_CACHE) / f"models--{repo_id.replace('/', '--')}" / "blobs"
return Path(constants.HF_HUB_CACHE) / f"models--{repo_id.replace('/', '--')}"
@staticmethod
def _cache_bytes(repo_id: str) -> int:
blobs = DiffusionBackend._hub_cache_repo_dir(repo_id) / "blobs"
total = 0
try:
for entry in blobs.iterdir():
@ -861,9 +866,7 @@ class DiffusionBackend:
local = Path(base).expanduser()
if local.is_dir():
return DiffusionBackend._local_dir_weight_bytes(local, exclude_transformer = True)
from huggingface_hub import constants
snapshots = Path(constants.HF_HUB_CACHE) / f"models--{base.replace('/', '--')}" / "snapshots"
snapshots = DiffusionBackend._hub_cache_repo_dir(base) / "snapshots"
if not snapshots.is_dir():
return 0
# Multiple revisions may be cached; the active one is the fullest, so take the max.