mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-23 09:09:01 +03:00
771779849b
* migrate imports * migrate stable diffusion adapter tests * up * move last remaining bits. * simplify test policy wording * move around * move flux ip adapter testing stuff to flux module of test * fix(ci): harden GitHub Actions workflows (#14669) (#14787) fix(ci): harden workflow files flagged on #14669 Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com> * clarify comment. * remove name from batch_input --------- Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from .cache import (
|
|
CacheTesterMixin,
|
|
FasterCacheTesterMixin,
|
|
FirstBlockCacheTesterMixin,
|
|
MagCacheTesterMixin,
|
|
PyramidAttentionBroadcastTesterMixin,
|
|
TaylorSeerCacheTesterMixin,
|
|
)
|
|
from .common import BasePipelineTesterConfig, PipelineTesterMixin, assert_mean_pixel_difference, check_same_shape
|
|
from .from_pipe import FromPipeTesterMixin
|
|
from .ip_adapter import IPAdapterTesterMixin
|
|
from .lora import LoraMemoryTesterMixin, LoraTesterMixin, UNetLoraTesterMixin
|
|
from .memory import (
|
|
GroupOffloadTesterMixin,
|
|
LayerwiseCastingTesterMixin,
|
|
MemoryTesterMixin,
|
|
PipelineOffloadTesterMixin,
|
|
)
|
|
|
|
|
|
__all__ = [
|
|
"BasePipelineTesterConfig",
|
|
"PipelineTesterMixin",
|
|
"FromPipeTesterMixin",
|
|
"IPAdapterTesterMixin",
|
|
"LoraTesterMixin",
|
|
"LoraMemoryTesterMixin",
|
|
"UNetLoraTesterMixin",
|
|
"MemoryTesterMixin",
|
|
"PipelineOffloadTesterMixin",
|
|
"GroupOffloadTesterMixin",
|
|
"LayerwiseCastingTesterMixin",
|
|
"CacheTesterMixin",
|
|
"PyramidAttentionBroadcastTesterMixin",
|
|
"FasterCacheTesterMixin",
|
|
"FirstBlockCacheTesterMixin",
|
|
"TaylorSeerCacheTesterMixin",
|
|
"MagCacheTesterMixin",
|
|
"assert_mean_pixel_difference",
|
|
"check_same_shape",
|
|
]
|