library_name: diffusers | |
UNet part of diffusion model provided to generate 3D polycrystal structures.\ | |
Generation sample: | |
![image/png](https://cdn-uploads.huggingface.co/production/uploads/6497d9c28f9988c8337dd7ed/XqETehQs0oci7ZNVoS8X7.png) | |
Project repo: https://github.com/evgmaslov/structure_generation \ | |
Train code: https://github.com/evgmaslov/structure_generation/blob/main/train.ipynb | |