Skip to content

Commit 51b4318

Browse files
author
toilaluan
committed
allow special cache ids only
1 parent 7b4ad2d commit 51b4318

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/hooks/taylorseer_cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ def apply_taylorseer_cache(module: torch.nn.Module, config: TaylorSeerCacheConfi
273273
logger.debug(f"Special cache identifiers: {special_cache_identifiers}")
274274

275275
for name, submodule in module.named_modules():
276-
if skip_compute_identifiers and special_cache_identifiers:
276+
if (skip_compute_identifiers and special_cache_identifiers) or (special_cache_identifiers):
277277
if any(re.fullmatch(identifier, name) for identifier in skip_compute_identifiers) or any(
278278
re.fullmatch(identifier, name) for identifier in special_cache_identifiers
279279
):

0 commit comments

Comments
 (0)