flowae/trainers/__init__.py

from .trainers import register, trainers_dict
from . import base_trainer
from . import ldm_trainer
from . import audio_ldm_trainer