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-06-26 12:36:45 +00:00
commit ca52807680
4 changed files with 88 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

9
studio/backend/core/inference/diffusion_cache.py
View file

@ -72,19 +72,20 @@ def apply_step_cache(
transformer = getattr(pipe, "transformer", None)
if transformer is None:
return None
thr = threshold if threshold is not None else (
QUANT_FBCACHE_THRESHOLD if quant_active else DEFAULT_FBCACHE_THRESHOLD
thr = (
threshold
if threshold is not None
else (QUANT_FBCACHE_THRESHOLD if quant_active else DEFAULT_FBCACHE_THRESHOLD)
)
try:
from diffusers import FirstBlockCacheConfig
config = FirstBlockCacheConfig(threshold=thr)
config = FirstBlockCacheConfig(threshold = thr)
enable_cache = getattr(transformer, "enable_cache", None)
if callable(enable_cache):
enable_cache(config)
else:
from diffusers.hooks import apply_first_block_cache
apply_first_block_cache(transformer, config)
try:
transformer._unsloth_step_cache = f"{mode}@{thr}"