gitextract_gx2sgdse/ ├── .gitignore ├── LICENSE ├── __init__.py ├── 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 ├── 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