custum_3d_diffusion/modules.py

__modules__ = {}

def register(name):
    def decorator(cls):
        __modules__[name] = cls
        return cls

    return decorator


def find(name):
    return __modules__[name]

from custum_3d_diffusion.trainings import base, image2mvimage_trainer, image2image_trainer