├── LICENSE ├── README.md ├── assets ├── demo_pic_sd35.png ├── demo_pic_z.png └── dmdr_method.png ├── requirements.txt └── train_cc └── sit ├── README.md ├── arguments.py ├── configs ├── default.yaml ├── multi_node.yaml └── z2.json ├── convert_weight ├── final_model.py ├── repa.py ├── repa_utils.py └── sit_conver.py ├── guidance_units.py ├── models_sit.py ├── samplers.py ├── scripts ├── cold_start_sitxl.sh └── dmdr_sitxl_refl.sh ├── train_sitxl_refl_deepspeed.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/README.md -------------------------------------------------------------------------------- /assets/demo_pic_sd35.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/assets/demo_pic_sd35.png -------------------------------------------------------------------------------- /assets/demo_pic_z.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/assets/demo_pic_z.png -------------------------------------------------------------------------------- /assets/dmdr_method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/assets/dmdr_method.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_cc/sit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/README.md -------------------------------------------------------------------------------- /train_cc/sit/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/arguments.py -------------------------------------------------------------------------------- /train_cc/sit/configs/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/configs/default.yaml -------------------------------------------------------------------------------- /train_cc/sit/configs/multi_node.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/configs/multi_node.yaml -------------------------------------------------------------------------------- /train_cc/sit/configs/z2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/configs/z2.json -------------------------------------------------------------------------------- /train_cc/sit/convert_weight/final_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/convert_weight/final_model.py -------------------------------------------------------------------------------- /train_cc/sit/convert_weight/repa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/convert_weight/repa.py -------------------------------------------------------------------------------- /train_cc/sit/convert_weight/repa_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/convert_weight/repa_utils.py -------------------------------------------------------------------------------- /train_cc/sit/convert_weight/sit_conver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/convert_weight/sit_conver.py -------------------------------------------------------------------------------- /train_cc/sit/guidance_units.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/guidance_units.py -------------------------------------------------------------------------------- /train_cc/sit/models_sit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/models_sit.py -------------------------------------------------------------------------------- /train_cc/sit/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/samplers.py -------------------------------------------------------------------------------- /train_cc/sit/scripts/cold_start_sitxl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/scripts/cold_start_sitxl.sh -------------------------------------------------------------------------------- /train_cc/sit/scripts/dmdr_sitxl_refl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/scripts/dmdr_sitxl_refl.sh -------------------------------------------------------------------------------- /train_cc/sit/train_sitxl_refl_deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/train_sitxl_refl_deepspeed.py -------------------------------------------------------------------------------- /train_cc/sit/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvvvjdy/dmdr/HEAD/train_cc/sit/utils.py --------------------------------------------------------------------------------