mirror of https://github.com/vladmandic/automatic
16 lines
437 B
Python
16 lines
437 B
Python
"""
|
|
This module contains a collection of U-Net models.
|
|
The :mod:`cr.models.unets` module includes the following classes:
|
|
|
|
- :class:`DiffusersUNet2DWrapper`: A 2D U-Net model for diffusers.
|
|
- :class:`DiffusersUNet2DCondWrapper`: A 2D U-Net model for diffusers with conditional input.
|
|
"""
|
|
|
|
from .unet import DiffusersUNet2DCondWrapper, DiffusersUNet2DWrapper
|
|
|
|
|
|
__all__ = [
|
|
"DiffusersUNet2DWrapper",
|
|
"DiffusersUNet2DCondWrapper",
|
|
]
|