from utils.interpolation_modules.blocks import VAEBlock, DiTBlock