Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 369 Bytes
Newer Older
Patrick von Platen's avatar
init upload  
Patrick von Platen committed
1
2
3
4
5
6 
# flake8: noqa
# There's no way to ignore "F401 '...' imported but unused" warnings in this
# module, but to preserve other warnings. So, don't check this module at all.

__version__ = "0.0.1"
Patrick von Platen's avatar
add pretrained model and pretrained sampler  
Patrick von Platen committed
7
8 
from .models.unet import UNetModel
from .samplers.gaussian import GaussianDiffusion
Patrick von Platen's avatar
add pipeline  
Patrick von Platen committed
9
10
11 

from .pipeline_utils import DiffusionPipeline
from .modeling_utils import PreTrainedModel