├── .gitignore ├── LICENSE ├── README.md ├── configs ├── sd2_baseline.yaml ├── sd2_ladd.yaml └── sd2_shortcut.yaml ├── flow_models ├── __init__.py ├── data │ └── transforms.py ├── distances.py ├── generation.py ├── inversion.py ├── loops.py ├── losses.py └── utils.py ├── images ├── kitten.png ├── puppy.png ├── puppy_rec.png ├── recflow_1_5_steps.png ├── recflow_2_5_steps.png ├── rectflow_1_1_step.png ├── rectflow_2_1_step.png ├── rectflow_2_distill_1_step_midpoint.png ├── rectflow_2_distill_1step.png ├── rectflow_2_distill_4_steps_ladd.png ├── shortcut.png └── text2img.png ├── requirements.txt ├── sd_2_fm_finetuning.ipynb ├── sd_2_fm_finetuning_immiscible.ipynb ├── sd_2_fm_finetuning_ladd.ipynb ├── sd_2_fm_finetuning_reflow_k.ipynb ├── sd_2_fm_finetuning_shortcut.ipynb ├── sd_2_fm_inference.ipynb └── sd_2_fm_inversion.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/README.md -------------------------------------------------------------------------------- /configs/sd2_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/configs/sd2_baseline.yaml -------------------------------------------------------------------------------- /configs/sd2_ladd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/configs/sd2_ladd.yaml -------------------------------------------------------------------------------- /configs/sd2_shortcut.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/configs/sd2_shortcut.yaml -------------------------------------------------------------------------------- /flow_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/__init__.py -------------------------------------------------------------------------------- /flow_models/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/data/transforms.py -------------------------------------------------------------------------------- /flow_models/distances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/distances.py -------------------------------------------------------------------------------- /flow_models/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/generation.py -------------------------------------------------------------------------------- /flow_models/inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/inversion.py -------------------------------------------------------------------------------- /flow_models/loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/loops.py -------------------------------------------------------------------------------- /flow_models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/losses.py -------------------------------------------------------------------------------- /flow_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/flow_models/utils.py -------------------------------------------------------------------------------- /images/kitten.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/kitten.png -------------------------------------------------------------------------------- /images/puppy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/puppy.png -------------------------------------------------------------------------------- /images/puppy_rec.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/puppy_rec.png -------------------------------------------------------------------------------- /images/recflow_1_5_steps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/recflow_1_5_steps.png -------------------------------------------------------------------------------- /images/recflow_2_5_steps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/recflow_2_5_steps.png -------------------------------------------------------------------------------- /images/rectflow_1_1_step.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/rectflow_1_1_step.png -------------------------------------------------------------------------------- /images/rectflow_2_1_step.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/rectflow_2_1_step.png -------------------------------------------------------------------------------- /images/rectflow_2_distill_1_step_midpoint.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/rectflow_2_distill_1_step_midpoint.png -------------------------------------------------------------------------------- /images/rectflow_2_distill_1step.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/rectflow_2_distill_1step.png -------------------------------------------------------------------------------- /images/rectflow_2_distill_4_steps_ladd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/rectflow_2_distill_4_steps_ladd.png -------------------------------------------------------------------------------- /images/shortcut.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/shortcut.png -------------------------------------------------------------------------------- /images/text2img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/images/text2img.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/requirements.txt -------------------------------------------------------------------------------- /sd_2_fm_finetuning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_finetuning.ipynb -------------------------------------------------------------------------------- /sd_2_fm_finetuning_immiscible.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_finetuning_immiscible.ipynb -------------------------------------------------------------------------------- /sd_2_fm_finetuning_ladd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_finetuning_ladd.ipynb -------------------------------------------------------------------------------- /sd_2_fm_finetuning_reflow_k.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_finetuning_reflow_k.ipynb -------------------------------------------------------------------------------- /sd_2_fm_finetuning_shortcut.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_finetuning_shortcut.ipynb -------------------------------------------------------------------------------- /sd_2_fm_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_inference.ipynb -------------------------------------------------------------------------------- /sd_2_fm_inversion.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leffff/FlowModels/HEAD/sd_2_fm_inversion.ipynb --------------------------------------------------------------------------------