├── .gitignore ├── assets └── teaser.png ├── custom_data └── example.csv ├── inference_scripts ├── analysis.py ├── eval.py ├── eval_custom.sh ├── eval_dpg.sh └── eval_geneval.sh ├── outputs └── touch.txt ├── readme.MD ├── requirements.txt └── train_scripts ├── accelerate_config.json ├── debug_attention.py ├── dino_lib.py ├── dpg_util.py ├── geneval_utils.py ├── modeling ├── __init__.py ├── context_transformer.py ├── encoder.py ├── pipeline.py └── sana.py ├── test.py ├── test_eval.py ├── train_reflect_dit.sh └── train_sana_self_correction.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/.gitignore -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /custom_data/example.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/custom_data/example.csv -------------------------------------------------------------------------------- /inference_scripts/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/inference_scripts/analysis.py -------------------------------------------------------------------------------- /inference_scripts/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/inference_scripts/eval.py -------------------------------------------------------------------------------- /inference_scripts/eval_custom.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/inference_scripts/eval_custom.sh -------------------------------------------------------------------------------- /inference_scripts/eval_dpg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/inference_scripts/eval_dpg.sh -------------------------------------------------------------------------------- /inference_scripts/eval_geneval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/inference_scripts/eval_geneval.sh -------------------------------------------------------------------------------- /outputs/touch.txt: -------------------------------------------------------------------------------- 1 | # make sure folder is created -------------------------------------------------------------------------------- /readme.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/readme.MD -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_scripts/accelerate_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/accelerate_config.json -------------------------------------------------------------------------------- /train_scripts/debug_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/debug_attention.py -------------------------------------------------------------------------------- /train_scripts/dino_lib.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_scripts/dpg_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/dpg_util.py -------------------------------------------------------------------------------- /train_scripts/geneval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/geneval_utils.py -------------------------------------------------------------------------------- /train_scripts/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_scripts/modeling/context_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/modeling/context_transformer.py -------------------------------------------------------------------------------- /train_scripts/modeling/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/modeling/encoder.py -------------------------------------------------------------------------------- /train_scripts/modeling/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/modeling/pipeline.py -------------------------------------------------------------------------------- /train_scripts/modeling/sana.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/modeling/sana.py -------------------------------------------------------------------------------- /train_scripts/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/test.py -------------------------------------------------------------------------------- /train_scripts/test_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/test_eval.py -------------------------------------------------------------------------------- /train_scripts/train_reflect_dit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/train_reflect_dit.sh -------------------------------------------------------------------------------- /train_scripts/train_sana_self_correction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacklishufan/Reflect-DiT/HEAD/train_scripts/train_sana_self_correction.py --------------------------------------------------------------------------------