Skip to content

Commit 8915e3a

Browse files
committed
Revert "First Block Cache (huggingface#11180)"
This reverts commit 0454fbb
1 parent 38740dd commit 8915e3a

31 files changed

+171
-930
lines changed

docs/source/en/api/cache.md

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,3 @@ Cache methods speedup diffusion transformers by storing and reusing intermediate
2828
[[autodoc]] FasterCacheConfig
2929

3030
[[autodoc]] apply_faster_cache
31-
32-
### FirstBlockCacheConfig
33-
34-
[[autodoc]] FirstBlockCacheConfig
35-
36-
[[autodoc]] apply_first_block_cache

src/diffusers/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,6 @@
149149
_import_structure["hooks"].extend(
150150
[
151151
"FasterCacheConfig",
152-
"FirstBlockCacheConfig",
153152
"HookRegistry",
154153
"LayerSkipConfig",
155154
"PyramidAttentionBroadcastConfig",
@@ -813,7 +812,6 @@
813812
)
814813
from .hooks import (
815814
FasterCacheConfig,
816-
FirstBlockCacheConfig,
817815
HookRegistry,
818816
LayerSkipConfig,
819817
PyramidAttentionBroadcastConfig,

src/diffusers/hooks/__init__.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,8 @@
1-
# Copyright 2024 The HuggingFace Team. All rights reserved.
2-
#
3-
# Licensed under the Apache License, Version 2.0 (the "License");
4-
# you may not use this file except in compliance with the License.
5-
# You may obtain a copy of the License at
6-
#
7-
# http://www.apache.org/licenses/LICENSE-2.0
8-
#
9-
# Unless required by applicable law or agreed to in writing, software
10-
# distributed under the License is distributed on an "AS IS" BASIS,
11-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12-
# See the License for the specific language governing permissions and
13-
# limitations under the License.
14-
151
from ..utils import is_torch_available
162

173

184
if is_torch_available():
195
from .faster_cache import FasterCacheConfig, apply_faster_cache
20-
from .first_block_cache import FirstBlockCacheConfig, apply_first_block_cache
216
from .group_offloading import apply_group_offloading
227
from .hooks import HookRegistry, ModelHook
238
from .layer_skip import LayerSkipConfig, apply_layer_skip

src/diffusers/hooks/_helpers.py

Lines changed: 0 additions & 293 deletions
This file was deleted.

0 commit comments

Comments
 (0)