grnr9730's picture
Update custom_pipeline/__init__.py
1085220 verified
from .WanTransformer3DModel import WanTransformer3DModel
from .AutoencoderKLWan import AutoencoderKLWan
__all__ = ["WanTransformer3DModel", "AutoencoderKLWan"]