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-05-25 05:03:41 +00:00
commit d8b785a4e2
2 changed files with 8 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1052,7 +1052,9 @@ def _drain_cuda_cache() -> None:
mps_backend = getattr(getattr(torch, "backends", None), "mps", None)
if mps_backend is not None and mps_backend.is_available():
mps_module = getattr(torch, "mps", None)
empty_cache = getattr(mps_module, "empty_cache", None) if mps_module else None
empty_cache = (
getattr(mps_module, "empty_cache", None) if mps_module else None
)
if empty_cache is not None:
empty_cache()
except Exception:

22
studio/backend/routes/models.py
View file

@ -2007,14 +2007,8 @@ async def delete_finetuned_model(
# specific quant file. If the loaded pipeline uses a
# DIFFERENT variant from the same directory, the delete
# is safe. Round 12 review #3.
loaded_gguf = (
diff_status.get("gguf_filename") or ""
).lower()
wants_variant = (
export_type == "gguf"
and gguf_variant
and loaded_gguf
)
loaded_gguf = (diff_status.get("gguf_filename") or "").lower()
wants_variant = export_type == "gguf" and gguf_variant and loaded_gguf
for candidate in candidates:
try:
candidate_path = Path(candidate).expanduser()
@ -2039,9 +2033,7 @@ async def delete_finetuned_model(
# different quant than the loaded one.
if wants_variant:
variant_low = gguf_variant.lower()
loaded_label = (
_extract_quant_label(loaded_gguf) or ""
).lower()
loaded_label = (_extract_quant_label(loaded_gguf) or "").lower()
if loaded_label and loaded_label != variant_low:
continue
raise HTTPException(
@ -2795,14 +2787,10 @@ async def delete_cached_model(
# quant via ``_delete_gguf_variant_files``. If the
# loaded pipeline uses a DIFFERENT variant from the
# same repo, the delete is safe. Round 12 review #4.
loaded_gguf = (
diff_status.get("gguf_filename") or ""
).lower()
loaded_gguf = (diff_status.get("gguf_filename") or "").lower()
if variant and loaded_gguf:
variant_low = variant.lower()
loaded_label = (
_extract_quant_label(loaded_gguf) or ""
).lower()
loaded_label = (_extract_quant_label(loaded_gguf) or "").lower()
if loaded_label and loaded_label != variant_low:
# Different quant from the same repo -> allow.
pass