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: