jjuke-diffusion

Use diffusions for various models


Keywords
JJukE_Diffusion, JJukE-Diffusion
License
MIT
Install
pip install jjuke-diffusion==0.0.1.0

Documentation

Diffusion bases for other various models

Contents

.
|-- README.md
|-- __init__.py
|-- diffusion
|   |-- __init__.py
|   |-- common.py
|   |-- ddim.py
|   |-- ddpm.py
|   |-- diffusion_base.py
|   |-- gaussian_diffusion.py
|   |-- karras.py
|   `-- ldm.py
|-- unet
|   |-- __init__.py
|   |-- base_modules.py
|   |-- unet_base.py
|   `-- unet_modules.py
`-- unet_cond
    |-- __init__.py
    |-- attention.py
    |-- fp16_util.py
    |-- transformer.py
    |-- unet.py
    `-- unet_modules.py

To-do List

  • Test 1D U-Net
  • Implement Conditional U-Net
  • Test LDM U-Net
  • Enjoy it!