Elastic-forcing's picture
Add training and inference source code
2ea2835 verified
Raw History Blame Contribute Delete
181 Bytes
"""distribution matching distribution-matching model."""
from .distribution_matching import DistributionMatchingModel, WanKVMMD
__all__ = ["DistributionMatchingModel", "WanKVMMD"]