Duplicated from stepfun-ai/Step1X-3D
2ac1c2d
1
2
from . import shape_autoencoder, shape_diffusion, shape_rectified_flow