---
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 \
Inference code: https://github.com/evgmaslov/structure_generation/blob/main/inference.ipynb