├── Checkpoints ├── SD3_model │ └── see_huggingface.txt ├── dit │ └── see_huggingface.txt ├── dit_lora │ └── see_huggingface.txt ├── llm_lora │ └── see_huggingface.txt ├── llm_model │ └── see_huggingface.txt └── llm_model_preprocessor │ ├── added_tokens.json │ ├── chat_template.json │ ├── merges.txt │ ├── preprocessor_config.json │ ├── special_tokens_map.json │ ├── tokenizer.json │ ├── tokenizer_config.json │ └── vocab.json ├── README.md ├── Unified-Bench ├── CLIP.py ├── DINO_v2.py ├── DINO_v3.py ├── LongCLIP.py ├── Score_i2i.py ├── UniBench │ └── example_image │ │ ├── 0.png │ │ └── 37.png ├── cp_image.py └── results │ └── example.json ├── assets └── figure1_new.jpg ├── demo.py ├── requirements.txt └── uae ├── models ├── clip_score.py ├── modeling_longcontext.py ├── modeling_longcontext_rec.py └── sd_model.py └── utils └── denoiser_prompt_embeds.py /Checkpoints/SD3_model/see_huggingface.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Checkpoints/dit/see_huggingface.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Checkpoints/dit_lora/see_huggingface.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Checkpoints/llm_lora/see_huggingface.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Checkpoints/llm_model/see_huggingface.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/added_tokens.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/added_tokens.json -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/chat_template.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/chat_template.json -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/merges.txt -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/preprocessor_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/preprocessor_config.json -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/special_tokens_map.json -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/tokenizer.json -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/tokenizer_config.json -------------------------------------------------------------------------------- /Checkpoints/llm_model_preprocessor/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Checkpoints/llm_model_preprocessor/vocab.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/README.md -------------------------------------------------------------------------------- /Unified-Bench/CLIP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/CLIP.py -------------------------------------------------------------------------------- /Unified-Bench/DINO_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/DINO_v2.py -------------------------------------------------------------------------------- /Unified-Bench/DINO_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/DINO_v3.py -------------------------------------------------------------------------------- /Unified-Bench/LongCLIP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/LongCLIP.py -------------------------------------------------------------------------------- /Unified-Bench/Score_i2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/Score_i2i.py -------------------------------------------------------------------------------- /Unified-Bench/UniBench/example_image/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/UniBench/example_image/0.png -------------------------------------------------------------------------------- /Unified-Bench/UniBench/example_image/37.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/UniBench/example_image/37.png -------------------------------------------------------------------------------- /Unified-Bench/cp_image.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Unified-Bench/results/example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/Unified-Bench/results/example.json -------------------------------------------------------------------------------- /assets/figure1_new.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/assets/figure1_new.jpg -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/demo.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /uae/models/clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/uae/models/clip_score.py -------------------------------------------------------------------------------- /uae/models/modeling_longcontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/uae/models/modeling_longcontext.py -------------------------------------------------------------------------------- /uae/models/modeling_longcontext_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/uae/models/modeling_longcontext_rec.py -------------------------------------------------------------------------------- /uae/models/sd_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/uae/models/sd_model.py -------------------------------------------------------------------------------- /uae/utils/denoiser_prompt_embeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/UAE/HEAD/uae/utils/denoiser_prompt_embeds.py --------------------------------------------------------------------------------