FilesExpand file tree

main

/

__init__.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

9 lines (7 loc) · 439 Bytes

main

/

__init__.py

Copy path

Top

File metadata and controls

• Code
• Blame

9 lines (7 loc) · 439 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

from ..utils import is_torch_available

if is_torch_available():

from .faster_cache import FasterCacheConfig, apply_faster_cache

from .group_offloading import apply_group_offloading

from .hooks import HookRegistry, ModelHook

from .layerwise_casting import apply_layerwise_casting, apply_layerwise_casting_hook

from .pyramid_attention_broadcast import PyramidAttentionBroadcastConfig, apply_pyramid_attention_broadcast