library_name: diffusers | |
UNet part of diffusion model provided to generate 3D polycrystal structures.\ | |
Generation sample: | |
 | |
Project repo: https://github.com/evgmaslov/structure_generation \ | |
Train code: https://github.com/evgmaslov/structure_generation/blob/main/train.ipynb \ | |
Inference code: https://github.com/evgmaslov/structure_generation/blob/main/inference.ipynb | |