__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .inference import (init_model, sample_conditional_model,
                        sample_ddpm_model, sample_img2img_model,
                        sample_unconditional_model)
from .train import set_random_seed, train_model

__all__ = [
    'set_random_seed', 'train_model', 'init_model', 'sample_img2img_model',
    'sample_unconditional_model', 'sample_conditional_model',
    'sample_ddpm_model'
]