├── .gitignore ├── LICENSE ├── README.md ├── app.py ├── assets └── cover.png ├── demo_Condition_LoRA ├── annotation.csv ├── stylebooth_loader.py └── train_cond_lora.py ├── examples ├── car │ ├── 2532_A sturdy ATV with rugged looks..jpg │ ├── 2532_rank0_A sturdy ATV with rugged looks..json │ ├── depth.jpg │ ├── subject.jpg │ └── textual_guidance │ │ ├── rainy.jpg │ │ ├── snowy.jpg │ │ └── sunny.jpg ├── clipboard │ ├── 2462_Versatile clipboard with varied sizes and materials..jpg │ ├── 2462_rank0_Versatile clipboard with varied sizes and materials..json │ ├── canny.jpg │ ├── depth.jpg │ └── textual_guidance │ │ ├── Einstein.jpg │ │ ├── Mario.jpg │ │ ├── Word1.jpg │ │ └── Word2.jpg ├── doll │ ├── 1116_A spooky themed gothic doll..jpg │ ├── 1116_rank0_A spooky themed gothic doll..json │ ├── canny.jpg │ ├── subject.jpg │ └── textual_guidance │ │ ├── chinatown_pink.jpg │ │ ├── space_station_gold.jpg │ │ └── valley.jpg ├── guinea_pig │ ├── 1463_rank0_A brown and white fluffy guinea pig..json │ ├── canny.jpg │ ├── subject.jpg │ └── textual_guidance │ │ ├── hello_world.jpg │ │ └── mountain.jpg ├── spray │ ├── 701_A floral body spray.jpg │ ├── 701_rank0_A floral body spray.json │ ├── background.jpg │ └── subject.jpg ├── suitman │ ├── 1735_A traditional formal outfit for men..jpg │ ├── 1735_rank0_A traditional formal outfit for men..json │ ├── depth.jpg │ ├── subject.jpg │ └── textual_guidance │ │ ├── red.jpg │ │ └── white.jpg ├── toy │ ├── 1616_A soft, plush toy with cuddly features..jpg │ ├── 1616_rank0_A soft, plush toy with cuddly features..json │ ├── canny.jpg │ ├── depth.jpg │ └── textual_guidance │ │ ├── farm.jpg │ │ ├── night.jpg │ │ └── sunset.jpg └── window │ ├── 1634_A decorative fabric topper for windows..jpg │ ├── 1634_rank0_A decorative fabric topper for windows..json │ ├── background.jpg │ └── subject.jpg ├── inference.py ├── requirements.txt ├── src ├── UniCombinePipeline.py ├── UniCombineTransformer2DModel.py ├── UniCombineTransformerBlock.py ├── condition.py ├── dataloader.py ├── hook.py ├── lora_switching_module.py ├── partition_dataset.py └── text_encoder.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/app.py -------------------------------------------------------------------------------- /assets/cover.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/assets/cover.png -------------------------------------------------------------------------------- /demo_Condition_LoRA/annotation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/demo_Condition_LoRA/annotation.csv -------------------------------------------------------------------------------- /demo_Condition_LoRA/stylebooth_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/demo_Condition_LoRA/stylebooth_loader.py -------------------------------------------------------------------------------- /demo_Condition_LoRA/train_cond_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/demo_Condition_LoRA/train_cond_lora.py -------------------------------------------------------------------------------- /examples/car/2532_A sturdy ATV with rugged looks..jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/2532_A sturdy ATV with rugged looks..jpg -------------------------------------------------------------------------------- /examples/car/2532_rank0_A sturdy ATV with rugged looks..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/2532_rank0_A sturdy ATV with rugged looks..json -------------------------------------------------------------------------------- /examples/car/depth.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/depth.jpg -------------------------------------------------------------------------------- /examples/car/subject.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/subject.jpg -------------------------------------------------------------------------------- /examples/car/textual_guidance/rainy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/textual_guidance/rainy.jpg -------------------------------------------------------------------------------- /examples/car/textual_guidance/snowy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/textual_guidance/snowy.jpg -------------------------------------------------------------------------------- /examples/car/textual_guidance/sunny.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/car/textual_guidance/sunny.jpg -------------------------------------------------------------------------------- /examples/clipboard/2462_Versatile clipboard with varied sizes and materials..jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/2462_Versatile clipboard with varied sizes and materials..jpg -------------------------------------------------------------------------------- /examples/clipboard/2462_rank0_Versatile clipboard with varied sizes and materials..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/2462_rank0_Versatile clipboard with varied sizes and materials..json -------------------------------------------------------------------------------- /examples/clipboard/canny.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/canny.jpg -------------------------------------------------------------------------------- /examples/clipboard/depth.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/depth.jpg -------------------------------------------------------------------------------- /examples/clipboard/textual_guidance/Einstein.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/textual_guidance/Einstein.jpg -------------------------------------------------------------------------------- /examples/clipboard/textual_guidance/Mario.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/textual_guidance/Mario.jpg -------------------------------------------------------------------------------- /examples/clipboard/textual_guidance/Word1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/textual_guidance/Word1.jpg -------------------------------------------------------------------------------- /examples/clipboard/textual_guidance/Word2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/clipboard/textual_guidance/Word2.jpg -------------------------------------------------------------------------------- /examples/doll/1116_A spooky themed gothic doll..jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/1116_A spooky themed gothic doll..jpg -------------------------------------------------------------------------------- /examples/doll/1116_rank0_A spooky themed gothic doll..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/1116_rank0_A spooky themed gothic doll..json -------------------------------------------------------------------------------- /examples/doll/canny.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/canny.jpg -------------------------------------------------------------------------------- /examples/doll/subject.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/subject.jpg -------------------------------------------------------------------------------- /examples/doll/textual_guidance/chinatown_pink.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/textual_guidance/chinatown_pink.jpg -------------------------------------------------------------------------------- /examples/doll/textual_guidance/space_station_gold.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/textual_guidance/space_station_gold.jpg -------------------------------------------------------------------------------- /examples/doll/textual_guidance/valley.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/doll/textual_guidance/valley.jpg -------------------------------------------------------------------------------- /examples/guinea_pig/1463_rank0_A brown and white fluffy guinea pig..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/guinea_pig/1463_rank0_A brown and white fluffy guinea pig..json -------------------------------------------------------------------------------- /examples/guinea_pig/canny.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/guinea_pig/canny.jpg -------------------------------------------------------------------------------- /examples/guinea_pig/subject.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/guinea_pig/subject.jpg -------------------------------------------------------------------------------- /examples/guinea_pig/textual_guidance/hello_world.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/guinea_pig/textual_guidance/hello_world.jpg -------------------------------------------------------------------------------- /examples/guinea_pig/textual_guidance/mountain.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/guinea_pig/textual_guidance/mountain.jpg -------------------------------------------------------------------------------- /examples/spray/701_A floral body spray.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/spray/701_A floral body spray.jpg -------------------------------------------------------------------------------- /examples/spray/701_rank0_A floral body spray.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/spray/701_rank0_A floral body spray.json -------------------------------------------------------------------------------- /examples/spray/background.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/spray/background.jpg -------------------------------------------------------------------------------- /examples/spray/subject.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/spray/subject.jpg -------------------------------------------------------------------------------- /examples/suitman/1735_A traditional formal outfit for men..jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/suitman/1735_A traditional formal outfit for men..jpg -------------------------------------------------------------------------------- /examples/suitman/1735_rank0_A traditional formal outfit for men..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/suitman/1735_rank0_A traditional formal outfit for men..json -------------------------------------------------------------------------------- /examples/suitman/depth.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/suitman/depth.jpg -------------------------------------------------------------------------------- /examples/suitman/subject.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/suitman/subject.jpg -------------------------------------------------------------------------------- /examples/suitman/textual_guidance/red.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/suitman/textual_guidance/red.jpg -------------------------------------------------------------------------------- /examples/suitman/textual_guidance/white.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/suitman/textual_guidance/white.jpg -------------------------------------------------------------------------------- /examples/toy/1616_A soft, plush toy with cuddly features..jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/1616_A soft, plush toy with cuddly features..jpg -------------------------------------------------------------------------------- /examples/toy/1616_rank0_A soft, plush toy with cuddly features..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/1616_rank0_A soft, plush toy with cuddly features..json -------------------------------------------------------------------------------- /examples/toy/canny.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/canny.jpg -------------------------------------------------------------------------------- /examples/toy/depth.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/depth.jpg -------------------------------------------------------------------------------- /examples/toy/textual_guidance/farm.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/textual_guidance/farm.jpg -------------------------------------------------------------------------------- /examples/toy/textual_guidance/night.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/textual_guidance/night.jpg -------------------------------------------------------------------------------- /examples/toy/textual_guidance/sunset.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/toy/textual_guidance/sunset.jpg -------------------------------------------------------------------------------- /examples/window/1634_A decorative fabric topper for windows..jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/window/1634_A decorative fabric topper for windows..jpg -------------------------------------------------------------------------------- /examples/window/1634_rank0_A decorative fabric topper for windows..json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/window/1634_rank0_A decorative fabric topper for windows..json -------------------------------------------------------------------------------- /examples/window/background.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/window/background.jpg -------------------------------------------------------------------------------- /examples/window/subject.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/examples/window/subject.jpg -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/inference.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/UniCombinePipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/UniCombinePipeline.py -------------------------------------------------------------------------------- /src/UniCombineTransformer2DModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/UniCombineTransformer2DModel.py -------------------------------------------------------------------------------- /src/UniCombineTransformerBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/UniCombineTransformerBlock.py -------------------------------------------------------------------------------- /src/condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/condition.py -------------------------------------------------------------------------------- /src/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/dataloader.py -------------------------------------------------------------------------------- /src/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/hook.py -------------------------------------------------------------------------------- /src/lora_switching_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/lora_switching_module.py -------------------------------------------------------------------------------- /src/partition_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/partition_dataset.py -------------------------------------------------------------------------------- /src/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/src/text_encoder.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xuan-World/UniCombine/HEAD/train.py --------------------------------------------------------------------------------