__init__.py

import os

os.environ["NCCL_DEBUG"] = "ERROR"

from .diffusion.scheduler import *
from .diffusion.video_pipeline import *
from .modules.model import *