mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-23 09:09:01 +03:00
d30c748f5f
* tests: add lora tester mixin for modular and migrate minimax h3. * enable lora testing modular ci. * remove seed. * up
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from .common import (
|
|
BaseModularPipelineOutputMixin,
|
|
BaseModularPipelineTesterConfig,
|
|
ModularPipelineTesterMixin,
|
|
)
|
|
from .guider import ModularGuiderTesterMixin
|
|
from .loading import ModularLoadingTesterMixin
|
|
from .lora import ModularLoraMemoryTesterMixin, ModularLoraTesterMixin
|
|
from .memory import (
|
|
ModularAutoOffloadTesterMixin,
|
|
ModularGroupOffloadTesterMixin,
|
|
ModularMemoryTesterMixin,
|
|
ModularOffloadTesterMixin,
|
|
)
|
|
from .utils import backend_memory_allocated, get_specified_components, patch_free_memory
|
|
from .workflow import ModularWorkflowTesterMixin
|
|
|
|
|
|
__all__ = [
|
|
"BaseModularPipelineOutputMixin",
|
|
"BaseModularPipelineTesterConfig",
|
|
"ModularAutoOffloadTesterMixin",
|
|
"ModularGroupOffloadTesterMixin",
|
|
"ModularGuiderTesterMixin",
|
|
"ModularLoadingTesterMixin",
|
|
"ModularLoraMemoryTesterMixin",
|
|
"ModularLoraTesterMixin",
|
|
"ModularMemoryTesterMixin",
|
|
"ModularOffloadTesterMixin",
|
|
"ModularPipelineTesterMixin",
|
|
"ModularWorkflowTesterMixin",
|
|
"backend_memory_allocated",
|
|
"get_specified_components",
|
|
"patch_free_memory",
|
|
]
|