├── .gitignore ├── LICENSE ├── README.md ├── catsample.py ├── configs ├── config.yaml └── model │ ├── medium.yaml │ └── small.yaml ├── data.py ├── environment.yml ├── graph_lib.py ├── load_model.py ├── losses.py ├── model ├── __init__.py ├── ema.py ├── fused_add_dropout_scale.py ├── rotary.py ├── transformer.py └── utils.py ├── noise_lib.py ├── run_sample.py ├── run_sample_cond.py ├── run_train.py ├── sampling.py ├── test.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/README.md -------------------------------------------------------------------------------- /catsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/catsample.py -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/model/medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/configs/model/medium.yaml -------------------------------------------------------------------------------- /configs/model/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/configs/model/small.yaml -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/data.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/environment.yml -------------------------------------------------------------------------------- /graph_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/graph_lib.py -------------------------------------------------------------------------------- /load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/load_model.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/losses.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | from .transformer import SEDD -------------------------------------------------------------------------------- /model/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/model/ema.py -------------------------------------------------------------------------------- /model/fused_add_dropout_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/model/fused_add_dropout_scale.py -------------------------------------------------------------------------------- /model/rotary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/model/rotary.py -------------------------------------------------------------------------------- /model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/model/transformer.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/model/utils.py -------------------------------------------------------------------------------- /noise_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/noise_lib.py -------------------------------------------------------------------------------- /run_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/run_sample.py -------------------------------------------------------------------------------- /run_sample_cond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/run_sample_cond.py -------------------------------------------------------------------------------- /run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/run_train.py -------------------------------------------------------------------------------- /sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/sampling.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/louaaron/Score-Entropy-Discrete-Diffusion/HEAD/utils.py --------------------------------------------------------------------------------