├── .gitignore ├── LICENSE ├── __init__.py ├── assets ├── ace_method │ ├── ace_refiner.webp │ ├── ace_refiner_process.webp │ └── samples_compare.webp ├── comfyui │ ├── ace_control.png │ ├── ace_element.png │ ├── ace_example.jpg │ └── ace_semantic.png ├── figures │ ├── icon.png │ └── teaser.png └── videos │ └── demo_chat.gif ├── chatbot ├── __init__.py ├── ace_inference.py ├── config │ ├── chatbot_ui.yaml │ └── models │ │ └── ace_0.6b_512.yaml ├── example.py ├── infer.py ├── run_gradio.py └── utils.py ├── config ├── inference_config │ ├── chatbot_ui.yaml │ └── models │ │ ├── ace_0.6b_1024.yaml │ │ ├── ace_0.6b_1024_refiner.yaml │ │ └── ace_0.6b_512.yaml └── train_config │ ├── ace_0.6b_1024_train.yaml │ └── ace_0.6b_512_train.yaml ├── examples ├── input_images │ ├── example0.webp │ ├── example1.webp │ ├── example2.webp │ └── example3.webp └── output_images │ ├── example0_0.6b_512.png │ ├── example1_0.6b_512.png │ ├── example2_0.6b_512.png │ └── example3_0.6b_512.png ├── modules ├── __init__.py ├── data │ ├── __init__.py │ └── dataset │ │ ├── __init__.py │ │ └── dataset.py ├── inference │ └── __init__.py ├── model │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── ace.py │ │ ├── layers.py │ │ └── pos_embed.py │ ├── diffusion │ │ ├── __init__.py │ │ ├── diffusions.py │ │ ├── samplers.py │ │ └── schedules.py │ ├── embedder │ │ ├── __init__.py │ │ └── embedder.py │ ├── network │ │ ├── __init__.py │ │ └── ldm_ace.py │ └── utils │ │ └── basic_utils.py └── solver │ ├── __init__.py │ └── ace_solver.py ├── readme.md ├── requirements.txt └── tools ├── run_inference.py └── run_train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/LICENSE -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/ace_method/ace_refiner.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/ace_method/ace_refiner.webp -------------------------------------------------------------------------------- /assets/ace_method/ace_refiner_process.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/ace_method/ace_refiner_process.webp -------------------------------------------------------------------------------- /assets/ace_method/samples_compare.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/ace_method/samples_compare.webp -------------------------------------------------------------------------------- /assets/comfyui/ace_control.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/comfyui/ace_control.png -------------------------------------------------------------------------------- /assets/comfyui/ace_element.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/comfyui/ace_element.png -------------------------------------------------------------------------------- /assets/comfyui/ace_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/comfyui/ace_example.jpg -------------------------------------------------------------------------------- /assets/comfyui/ace_semantic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/comfyui/ace_semantic.png -------------------------------------------------------------------------------- /assets/figures/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/figures/icon.png -------------------------------------------------------------------------------- /assets/figures/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/figures/teaser.png -------------------------------------------------------------------------------- /assets/videos/demo_chat.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/assets/videos/demo_chat.gif -------------------------------------------------------------------------------- /chatbot/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /chatbot/ace_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/ace_inference.py -------------------------------------------------------------------------------- /chatbot/config/chatbot_ui.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/config/chatbot_ui.yaml -------------------------------------------------------------------------------- /chatbot/config/models/ace_0.6b_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/config/models/ace_0.6b_512.yaml -------------------------------------------------------------------------------- /chatbot/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/example.py -------------------------------------------------------------------------------- /chatbot/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/infer.py -------------------------------------------------------------------------------- /chatbot/run_gradio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/run_gradio.py -------------------------------------------------------------------------------- /chatbot/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/chatbot/utils.py -------------------------------------------------------------------------------- /config/inference_config/chatbot_ui.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/config/inference_config/chatbot_ui.yaml -------------------------------------------------------------------------------- /config/inference_config/models/ace_0.6b_1024.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/config/inference_config/models/ace_0.6b_1024.yaml -------------------------------------------------------------------------------- /config/inference_config/models/ace_0.6b_1024_refiner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/config/inference_config/models/ace_0.6b_1024_refiner.yaml -------------------------------------------------------------------------------- /config/inference_config/models/ace_0.6b_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/config/inference_config/models/ace_0.6b_512.yaml -------------------------------------------------------------------------------- /config/train_config/ace_0.6b_1024_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/config/train_config/ace_0.6b_1024_train.yaml -------------------------------------------------------------------------------- /config/train_config/ace_0.6b_512_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/config/train_config/ace_0.6b_512_train.yaml -------------------------------------------------------------------------------- /examples/input_images/example0.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/input_images/example0.webp -------------------------------------------------------------------------------- /examples/input_images/example1.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/input_images/example1.webp -------------------------------------------------------------------------------- /examples/input_images/example2.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/input_images/example2.webp -------------------------------------------------------------------------------- /examples/input_images/example3.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/input_images/example3.webp -------------------------------------------------------------------------------- /examples/output_images/example0_0.6b_512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/output_images/example0_0.6b_512.png -------------------------------------------------------------------------------- /examples/output_images/example1_0.6b_512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/output_images/example1_0.6b_512.png -------------------------------------------------------------------------------- /examples/output_images/example2_0.6b_512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/output_images/example2_0.6b_512.png -------------------------------------------------------------------------------- /examples/output_images/example3_0.6b_512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/examples/output_images/example3_0.6b_512.png -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- 1 | from . import data, model, solver -------------------------------------------------------------------------------- /modules/data/__init__.py: -------------------------------------------------------------------------------- 1 | from . import dataset -------------------------------------------------------------------------------- /modules/data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/data/dataset/__init__.py -------------------------------------------------------------------------------- /modules/data/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/data/dataset/dataset.py -------------------------------------------------------------------------------- /modules/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/__init__.py -------------------------------------------------------------------------------- /modules/model/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/backbone/__init__.py -------------------------------------------------------------------------------- /modules/model/backbone/ace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/backbone/ace.py -------------------------------------------------------------------------------- /modules/model/backbone/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/backbone/layers.py -------------------------------------------------------------------------------- /modules/model/backbone/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/backbone/pos_embed.py -------------------------------------------------------------------------------- /modules/model/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/diffusion/__init__.py -------------------------------------------------------------------------------- /modules/model/diffusion/diffusions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/diffusion/diffusions.py -------------------------------------------------------------------------------- /modules/model/diffusion/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/diffusion/samplers.py -------------------------------------------------------------------------------- /modules/model/diffusion/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/diffusion/schedules.py -------------------------------------------------------------------------------- /modules/model/embedder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/embedder/__init__.py -------------------------------------------------------------------------------- /modules/model/embedder/embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/embedder/embedder.py -------------------------------------------------------------------------------- /modules/model/network/__init__.py: -------------------------------------------------------------------------------- 1 | from .ldm_ace import LdmACE -------------------------------------------------------------------------------- /modules/model/network/ldm_ace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/network/ldm_ace.py -------------------------------------------------------------------------------- /modules/model/utils/basic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/model/utils/basic_utils.py -------------------------------------------------------------------------------- /modules/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/solver/__init__.py -------------------------------------------------------------------------------- /modules/solver/ace_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/modules/solver/ace_solver.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/run_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/tools/run_inference.py -------------------------------------------------------------------------------- /tools/run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/ACE/HEAD/tools/run_train.py --------------------------------------------------------------------------------