├── .gitignore ├── LICENSE ├── code.txt └── dc_ar_models.txt ├── README.md ├── app.py ├── assets └── teaser.png ├── configs └── inference │ └── dc_ar_t2i_512.yaml ├── modeling ├── __init__.py ├── diffusion │ ├── __init__.py │ ├── diffloss.py │ ├── diffusion_utils.py │ ├── gaussian_diffusion.py │ └── respace.py ├── generator │ ├── __init__.py │ ├── dc_ar.py │ └── net │ │ ├── act.py │ │ ├── basic_modules.py │ │ ├── blocks.py │ │ ├── import_utils.py │ │ ├── norms.py │ │ └── utils.py ├── modules │ ├── __init__.py │ ├── base_model.py │ ├── blocks2.py │ ├── ema_model.py │ └── losses.py ├── tokenizer │ ├── configuration.py │ ├── dc_ht.py │ ├── networks │ │ ├── dc_ae.py │ │ └── dc_ae_blocks │ │ │ ├── act.py │ │ │ ├── norm.py │ │ │ ├── ops.py │ │ │ └── triton_rms_norm.py │ ├── quantizer │ │ ├── __init__.py │ │ ├── hybrid_vq.py │ │ └── quantizer.py │ └── utils │ │ ├── __init__.py │ │ ├── dist.py │ │ ├── init.py │ │ ├── list.py │ │ ├── metric.py │ │ ├── misc.py │ │ ├── network.py │ │ ├── random.py │ │ └── transforms.py └── utils.py ├── pyproject.toml ├── sample.py └── utils ├── __init__.py ├── demo_util.py └── safety_check.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE/code.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/LICENSE/code.txt -------------------------------------------------------------------------------- /LICENSE/dc_ar_models.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/LICENSE/dc_ar_models.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/app.py -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /configs/inference/dc_ar_t2i_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/configs/inference/dc_ar_t2i_512.yaml -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modeling/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/diffusion/__init__.py -------------------------------------------------------------------------------- /modeling/diffusion/diffloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/diffusion/diffloss.py -------------------------------------------------------------------------------- /modeling/diffusion/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/diffusion/diffusion_utils.py -------------------------------------------------------------------------------- /modeling/diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /modeling/diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/diffusion/respace.py -------------------------------------------------------------------------------- /modeling/generator/__init__.py: -------------------------------------------------------------------------------- 1 | from .dc_ar import DCAR -------------------------------------------------------------------------------- /modeling/generator/dc_ar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/dc_ar.py -------------------------------------------------------------------------------- /modeling/generator/net/act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/net/act.py -------------------------------------------------------------------------------- /modeling/generator/net/basic_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/net/basic_modules.py -------------------------------------------------------------------------------- /modeling/generator/net/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/net/blocks.py -------------------------------------------------------------------------------- /modeling/generator/net/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/net/import_utils.py -------------------------------------------------------------------------------- /modeling/generator/net/norms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/net/norms.py -------------------------------------------------------------------------------- /modeling/generator/net/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/generator/net/utils.py -------------------------------------------------------------------------------- /modeling/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/modules/__init__.py -------------------------------------------------------------------------------- /modeling/modules/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/modules/base_model.py -------------------------------------------------------------------------------- /modeling/modules/blocks2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/modules/blocks2.py -------------------------------------------------------------------------------- /modeling/modules/ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/modules/ema_model.py -------------------------------------------------------------------------------- /modeling/modules/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/modules/losses.py -------------------------------------------------------------------------------- /modeling/tokenizer/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/configuration.py -------------------------------------------------------------------------------- /modeling/tokenizer/dc_ht.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/dc_ht.py -------------------------------------------------------------------------------- /modeling/tokenizer/networks/dc_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/networks/dc_ae.py -------------------------------------------------------------------------------- /modeling/tokenizer/networks/dc_ae_blocks/act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/networks/dc_ae_blocks/act.py -------------------------------------------------------------------------------- /modeling/tokenizer/networks/dc_ae_blocks/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/networks/dc_ae_blocks/norm.py -------------------------------------------------------------------------------- /modeling/tokenizer/networks/dc_ae_blocks/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/networks/dc_ae_blocks/ops.py -------------------------------------------------------------------------------- /modeling/tokenizer/networks/dc_ae_blocks/triton_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/networks/dc_ae_blocks/triton_rms_norm.py -------------------------------------------------------------------------------- /modeling/tokenizer/quantizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/quantizer/__init__.py -------------------------------------------------------------------------------- /modeling/tokenizer/quantizer/hybrid_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/quantizer/hybrid_vq.py -------------------------------------------------------------------------------- /modeling/tokenizer/quantizer/quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/quantizer/quantizer.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/__init__.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/dist.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/init.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/list.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/metric.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/misc.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/network.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/random.py -------------------------------------------------------------------------------- /modeling/tokenizer/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/tokenizer/utils/transforms.py -------------------------------------------------------------------------------- /modeling/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/modeling/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/pyproject.toml -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/sample.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/demo_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/utils/demo_util.py -------------------------------------------------------------------------------- /utils/safety_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-AR/HEAD/utils/safety_check.py --------------------------------------------------------------------------------