Pixart-Sigma / diffusion /__init__.py
artificialguybr's picture
Hi
eadd7b4
# Modified from OpenAI's diffusion repos
# GLIDE: https://github.com/openai/glide-text2im/blob/main/glide_text2im/gaussian_diffusion.py
# ADM: https://github.com/openai/guided-diffusion/blob/main/guided_diffusion
# IDDPM: https://github.com/openai/improved-diffusion/blob/main/improved_diffusion/gaussian_diffusion.py
from .iddpm import IDDPM
from .dpm_solver import DPMS
from .sa_sampler import SASolverSampler