mirror of
https://github.com/Stability-AI/generative-models.git
synced 2025-12-24 08:44:20 +01:00
e596332148
* clean requirements * rm taming deps * isort, black * mv lipips, license * clean vq, fix path * fix loss path, gitignore * tested requirements pt13 * fix numpy req for python3.8, add tests * fix name * fix dep scipy 3.8 pt2 * add black test formatter
8 lines
268 B
Python
8 lines
268 B
Python
from .denoiser import Denoiser
|
|
from .discretizer import Discretization
|
|
from .loss import StandardDiffusionLoss
|
|
from .model import Decoder, Encoder, Model
|
|
from .openaimodel import UNetModel
|
|
from .sampling import BaseDiffusionSampler
|
|
from .wrappers import OpenAIWrapper
|