| datasets: | |
| - imagenet-1k | |
| language: | |
| - en | |
| license: mit | |
| pipeline_tag: unconditional-image-generation | |
| tags: | |
| - art | |
| # TerDiT | |
| This repository contains the trained model for the paper ["TerDiT: Ternary Diffusion Models with Transformers"](https://huggingface.co/papers/2405.14854) | |
| Code: https://github.com/Lucky-Lance/TerDiT | |
| 256x256 4.2B model: [TerDiT-4.2B](https://huggingface.co/lucky-lance/TerDiT/tree/main/3B_1180000) | |
| 256x256 600M model: [TerDiT-600M](https://huggingface.co/lucky-lance/TerDiT/tree/main/600M_1750000) | |
| 512x512 4.2B model: [TerDiT-4.2B](https://huggingface.co/lucky-lance/TerDiT/tree/main/3B_512_1900000) | |
| The codebase is extended from [Large-DiT-ImageNet](https://github.com/Alpha-VLLM/LLaMA2-Accessory/tree/main/Large-DiT-ImageNet) and highly motivated by [Lumina-T2X](https://github.com/Alpha-VLLM/Lumina-T2X). Thanks for their awesome work! |