├── .gitignore ├── LICENSE ├── README.md ├── model_data └── README.md ├── nets ├── __init__.py ├── attention.py ├── pipeline.py └── transformer_2d.py ├── predict.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /model_data/README.md: -------------------------------------------------------------------------------- 1 | Please put weights here. -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nets/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/nets/attention.py -------------------------------------------------------------------------------- /nets/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/nets/pipeline.py -------------------------------------------------------------------------------- /nets/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/nets/transformer_2d.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/DiT-pytorch/HEAD/requirements.txt --------------------------------------------------------------------------------