├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── app.py ├── assets ├── cat_cafe.png ├── clock.png ├── crystal_ball.png ├── cup.png ├── examples │ ├── 0_0.json │ ├── 0_0.png │ ├── 1_0.json │ ├── 1_0.png │ ├── 1one2one │ │ ├── config.json │ │ ├── ref1.jpg │ │ └── result.png │ ├── 2_0.json │ ├── 2_0.png │ ├── 2one2one │ │ ├── config.json │ │ ├── ref1.png │ │ └── result.png │ ├── 3two2one │ │ ├── config.json │ │ ├── ref1.png │ │ ├── ref2.png │ │ └── result.png │ ├── 4two2one │ │ ├── config.json │ │ ├── ref1.png │ │ ├── ref2.png │ │ └── result.png │ ├── 5many2one │ │ ├── config.json │ │ ├── ref1.png │ │ ├── ref2.png │ │ ├── ref3.png │ │ └── result.png │ └── 6t2i │ │ ├── config.json │ │ └── result.png ├── figurine.png ├── logo.png ├── simplecase.jpeg ├── simplecase.jpg ├── teaser.jpg ├── teaser_umo2uno.webp ├── teaser_uso2uno.webp └── uno1m.webp ├── config └── deepspeed │ ├── zero2_config.json │ └── zero3_config.json ├── datasets ├── dreambench_multiip.json ├── dreambench_singleip.json └── dreambench_toy.json ├── eval ├── evaluate_clip_dino_score_multi_subject.py └── evaluate_clip_dino_score_single_subject.py ├── inference.py ├── pyproject.toml ├── requirements.txt ├── template └── uno_instructions.py ├── train.py └── uno ├── dataset └── uno.py ├── flux ├── math.py ├── model.py ├── modules │ ├── autoencoder.py │ ├── conditioner.py │ └── layers.py ├── pipeline.py ├── sampling.py └── util.py └── utils ├── convert_yaml_to_args_file.py └── filter_uno_1m_dataset.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/app.py -------------------------------------------------------------------------------- /assets/cat_cafe.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/cat_cafe.png -------------------------------------------------------------------------------- /assets/clock.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/clock.png -------------------------------------------------------------------------------- /assets/crystal_ball.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/crystal_ball.png -------------------------------------------------------------------------------- /assets/cup.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/cup.png -------------------------------------------------------------------------------- /assets/examples/0_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/0_0.json -------------------------------------------------------------------------------- /assets/examples/0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/0_0.png -------------------------------------------------------------------------------- /assets/examples/1_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/1_0.json -------------------------------------------------------------------------------- /assets/examples/1_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/1_0.png -------------------------------------------------------------------------------- /assets/examples/1one2one/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/1one2one/config.json -------------------------------------------------------------------------------- /assets/examples/1one2one/ref1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/1one2one/ref1.jpg -------------------------------------------------------------------------------- /assets/examples/1one2one/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/1one2one/result.png -------------------------------------------------------------------------------- /assets/examples/2_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/2_0.json -------------------------------------------------------------------------------- /assets/examples/2_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/2_0.png -------------------------------------------------------------------------------- /assets/examples/2one2one/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/2one2one/config.json -------------------------------------------------------------------------------- /assets/examples/2one2one/ref1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/2one2one/ref1.png -------------------------------------------------------------------------------- /assets/examples/2one2one/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/2one2one/result.png -------------------------------------------------------------------------------- /assets/examples/3two2one/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/3two2one/config.json -------------------------------------------------------------------------------- /assets/examples/3two2one/ref1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/3two2one/ref1.png -------------------------------------------------------------------------------- /assets/examples/3two2one/ref2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/3two2one/ref2.png -------------------------------------------------------------------------------- /assets/examples/3two2one/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/3two2one/result.png -------------------------------------------------------------------------------- /assets/examples/4two2one/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/4two2one/config.json -------------------------------------------------------------------------------- /assets/examples/4two2one/ref1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/4two2one/ref1.png -------------------------------------------------------------------------------- /assets/examples/4two2one/ref2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/4two2one/ref2.png -------------------------------------------------------------------------------- /assets/examples/4two2one/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/4two2one/result.png -------------------------------------------------------------------------------- /assets/examples/5many2one/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/5many2one/config.json -------------------------------------------------------------------------------- /assets/examples/5many2one/ref1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/5many2one/ref1.png -------------------------------------------------------------------------------- /assets/examples/5many2one/ref2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/5many2one/ref2.png -------------------------------------------------------------------------------- /assets/examples/5many2one/ref3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/5many2one/ref3.png -------------------------------------------------------------------------------- /assets/examples/5many2one/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/5many2one/result.png -------------------------------------------------------------------------------- /assets/examples/6t2i/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/6t2i/config.json -------------------------------------------------------------------------------- /assets/examples/6t2i/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/examples/6t2i/result.png -------------------------------------------------------------------------------- /assets/figurine.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/figurine.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/logo.png -------------------------------------------------------------------------------- /assets/simplecase.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/simplecase.jpeg -------------------------------------------------------------------------------- /assets/simplecase.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/simplecase.jpg -------------------------------------------------------------------------------- /assets/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/teaser.jpg -------------------------------------------------------------------------------- /assets/teaser_umo2uno.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/teaser_umo2uno.webp -------------------------------------------------------------------------------- /assets/teaser_uso2uno.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/teaser_uso2uno.webp -------------------------------------------------------------------------------- /assets/uno1m.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/assets/uno1m.webp -------------------------------------------------------------------------------- /config/deepspeed/zero2_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/config/deepspeed/zero2_config.json -------------------------------------------------------------------------------- /config/deepspeed/zero3_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/config/deepspeed/zero3_config.json -------------------------------------------------------------------------------- /datasets/dreambench_multiip.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/datasets/dreambench_multiip.json -------------------------------------------------------------------------------- /datasets/dreambench_singleip.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/datasets/dreambench_singleip.json -------------------------------------------------------------------------------- /datasets/dreambench_toy.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/datasets/dreambench_toy.json -------------------------------------------------------------------------------- /eval/evaluate_clip_dino_score_multi_subject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/eval/evaluate_clip_dino_score_multi_subject.py -------------------------------------------------------------------------------- /eval/evaluate_clip_dino_score_single_subject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/eval/evaluate_clip_dino_score_single_subject.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/inference.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/requirements.txt -------------------------------------------------------------------------------- /template/uno_instructions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/template/uno_instructions.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/train.py -------------------------------------------------------------------------------- /uno/dataset/uno.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/dataset/uno.py -------------------------------------------------------------------------------- /uno/flux/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/math.py -------------------------------------------------------------------------------- /uno/flux/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/model.py -------------------------------------------------------------------------------- /uno/flux/modules/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/modules/autoencoder.py -------------------------------------------------------------------------------- /uno/flux/modules/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/modules/conditioner.py -------------------------------------------------------------------------------- /uno/flux/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/modules/layers.py -------------------------------------------------------------------------------- /uno/flux/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/pipeline.py -------------------------------------------------------------------------------- /uno/flux/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/sampling.py -------------------------------------------------------------------------------- /uno/flux/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/flux/util.py -------------------------------------------------------------------------------- /uno/utils/convert_yaml_to_args_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/utils/convert_yaml_to_args_file.py -------------------------------------------------------------------------------- /uno/utils/filter_uno_1m_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/UNO/HEAD/uno/utils/filter_uno_1m_dataset.py --------------------------------------------------------------------------------