Files
FiDA-3D-Trellis/trellis/models/structured_latent_flow.py