diffusers/tests/models/testing_utils/__init__.py at cbb18786352e5693814c4098e2b2f0f6353b8d9a · huggingface/diffusers · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
from .attention import AttentionBackendTesterMixin, AttentionTesterMixin
from .cache import (
    CacheTesterMixin,
    FasterCacheConfigMixin,
    FasterCacheTesterMixin,
    FirstBlockCacheConfigMixin,
    FirstBlockCacheTesterMixin,
    PyramidAttentionBroadcastConfigMixin,
    PyramidAttentionBroadcastTesterMixin,
)
from .common import BaseModelTesterConfig, ModelTesterMixin
from .compile import TorchCompileTesterMixin
from .ip_adapter import IPAdapterTesterMixin
from .lora import LoraHotSwappingForModelTesterMixin, LoraTesterMixin
from .memory import CPUOffloadTesterMixin, GroupOffloadTesterMixin, LayerwiseCastingTesterMixin, MemoryTesterMixin
from .parallelism import ContextParallelAttentionBackendsTesterMixin, ContextParallelTesterMixin
from .quantization import (
    BitsAndBytesCompileTesterMixin,
    BitsAndBytesConfigMixin,
    BitsAndBytesTesterMixin,
    GGUFCompileTesterMixin,
    GGUFConfigMixin,
    GGUFTesterMixin,
    ModelOptCompileTesterMixin,
    ModelOptConfigMixin,
    ModelOptTesterMixin,
    QuantizationCompileTesterMixin,
    QuantizationTesterMixin,
    QuantoCompileTesterMixin,
    QuantoConfigMixin,
    QuantoTesterMixin,
    TorchAoCompileTesterMixin,
    TorchAoConfigMixin,
    TorchAoTesterMixin,
)
from .single_file import SingleFileTesterMixin
from .training import TrainingTesterMixin


__all__ = [
    "AttentionBackendTesterMixin",
    "AttentionTesterMixin",
    "BaseModelTesterConfig",
    "BitsAndBytesCompileTesterMixin",
    "BitsAndBytesConfigMixin",
    "BitsAndBytesTesterMixin",
    "CacheTesterMixin",
    "ContextParallelTesterMixin",
    "ContextParallelAttentionBackendsTesterMixin",
    "CPUOffloadTesterMixin",
    "FasterCacheConfigMixin",
    "FasterCacheTesterMixin",
    "FirstBlockCacheConfigMixin",
    "FirstBlockCacheTesterMixin",
    "GGUFCompileTesterMixin",
    "GGUFConfigMixin",
    "GGUFTesterMixin",
    "GroupOffloadTesterMixin",
    "IPAdapterTesterMixin",
    "LayerwiseCastingTesterMixin",
    "LoraHotSwappingForModelTesterMixin",
    "LoraTesterMixin",
    "MemoryTesterMixin",
    "ModelOptCompileTesterMixin",
    "ModelOptConfigMixin",
    "ModelOptTesterMixin",
    "ModelTesterMixin",
    "PyramidAttentionBroadcastConfigMixin",
    "PyramidAttentionBroadcastTesterMixin",
    "QuantizationCompileTesterMixin",
    "QuantizationTesterMixin",
    "QuantoCompileTesterMixin",
    "QuantoConfigMixin",
    "QuantoTesterMixin",
    "SingleFileTesterMixin",
    "TorchAoCompileTesterMixin",
    "TorchAoConfigMixin",
    "TorchAoTesterMixin",
    "TorchCompileTesterMixin",
    "TrainingTesterMixin",
]