λ X , λ Y , λ Z represents weights for atom type, lattice matrix, and atom coordinate, respectively, with each weight set to 1.0. Model training was performed using a single A100 80GB GPU with the same batch size and learning rate scheduler as mentioned in the contrastive learning, except for the learning rate 10 - 3 for diffusion models.