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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-05 08:05:15 +00:00
commit 6bbd37f33f
2 changed files with 6 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
studio/backend/core/inference/video.py
View file

@ -131,9 +131,7 @@ def _is_trusted_video_repo(repo_id: str) -> bool:
def _detect_load_family(
repo_id: str,
gguf_filename: Optional[str],
family_override: Optional[str],
repo_id: str, gguf_filename: Optional[str], family_override: Optional[str]
) -> Optional[VideoFamily]:
"""Family detection shared by validate_load_request and the load worker: the
repo id first, then the picked filename -- a local directory or generically
@ -759,9 +757,7 @@ class VideoBackend:
mib_per_gb = 1000.0**3 / (1024.0 * 1024.0)
if kind == "pipeline":
model_dense_mib = (
int(sum(components) * mib_per_gb * dtype_scale)
if components is not None
else None
int(sum(components) * mib_per_gb * dtype_scale) if components is not None else None
)
companion_mib = None
else:

6
studio/backend/core/training/diffusion_train_common.py
View file

@ -597,7 +597,7 @@ def _plan_cache_variants(
# fallback. Over this budget the default falls back to per-step VAE encoding. A fixed
# constant (rather than a psutil RAM fraction) keeps the gate dependency-free and identical
# across hosts; it is deliberately conservative, well under a typical training host's RAM.
_LATENT_CACHE_BUDGET_BYTES = 4 * 1024 ** 3 # 4 GiB
_LATENT_CACHE_BUDGET_BYTES = 4 * 1024**3 # 4 GiB
# Returned by the cache builders when the estimated cache exceeds the budget: the caller
# keeps the VAE resident and encodes each step's latents in-loop. A distinct sentinel from
@ -614,7 +614,9 @@ def _latent_cache_forced() -> bool:
def _latent_cache_over_budget(
per_variant_bytes: int, total_variants: int, budget_bytes: Optional[int] = None
per_variant_bytes: int,
total_variants: int,
budget_bytes: Optional[int] = None,
) -> bool:
"""True when a cache of ``total_variants`` entries, each two fp32 tensors totalling
``per_variant_bytes``, is estimated to exceed ``budget_bytes``. ``per_variant_bytes`` is