├── LICENSE ├── README.md ├── baselines ├── targets_consistency_distillation.py ├── targets_consistency_training.py ├── targets_livereflow.py ├── targets_naive.py └── targets_progressive.py ├── data ├── fig-method5.png ├── fig-showcase4.png └── imagenet_labels.txt ├── environment.yml ├── helper_eval.py ├── helper_inference.py ├── math_utils.py ├── model.py ├── requirements.txt ├── targets_shortcut.py ├── train.py └── utils ├── __pycache__ ├── checkpoint.cpython-310.pyc ├── dataset.cpython-310.pyc ├── datasets.cpython-310.pyc ├── evaluation.cpython-310.pyc ├── evaluation.cpython-39.pyc ├── fid.cpython-310.pyc ├── pretrained_resnet.cpython-310.pyc ├── pretrained_utils.cpython-310.pyc ├── sharding.cpython-310.pyc ├── spectral_optimizer.cpython-310.pyc ├── stable_vae.cpython-310.pyc ├── train_state.cpython-310.pyc ├── typing.cpython-310.pyc ├── utils.cpython-310.pyc ├── utils.cpython-39.pyc ├── wandb.cpython-310.pyc └── wandb.cpython-39.pyc ├── checkpoint.py ├── datasets.py ├── fid.py ├── pretrained_resnet.py ├── sharding.py ├── stable_vae.py ├── train_state.py └── wandb.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/README.md -------------------------------------------------------------------------------- /baselines/targets_consistency_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/baselines/targets_consistency_distillation.py -------------------------------------------------------------------------------- /baselines/targets_consistency_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/baselines/targets_consistency_training.py -------------------------------------------------------------------------------- /baselines/targets_livereflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/baselines/targets_livereflow.py -------------------------------------------------------------------------------- /baselines/targets_naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/baselines/targets_naive.py -------------------------------------------------------------------------------- /baselines/targets_progressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/baselines/targets_progressive.py -------------------------------------------------------------------------------- /data/fig-method5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/data/fig-method5.png -------------------------------------------------------------------------------- /data/fig-showcase4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/data/fig-showcase4.png -------------------------------------------------------------------------------- /data/imagenet_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/data/imagenet_labels.txt -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/environment.yml -------------------------------------------------------------------------------- /helper_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/helper_eval.py -------------------------------------------------------------------------------- /helper_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/helper_inference.py -------------------------------------------------------------------------------- /math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/math_utils.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/requirements.txt -------------------------------------------------------------------------------- /targets_shortcut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/targets_shortcut.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/train.py -------------------------------------------------------------------------------- /utils/__pycache__/checkpoint.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/checkpoint.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dataset.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/dataset.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/datasets.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/datasets.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/evaluation.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/evaluation.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/evaluation.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/evaluation.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/fid.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/fid.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/pretrained_resnet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/pretrained_resnet.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/pretrained_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/pretrained_utils.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/sharding.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/sharding.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/spectral_optimizer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/spectral_optimizer.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/stable_vae.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/stable_vae.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/train_state.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/train_state.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/typing.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/typing.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/utils.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/wandb.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/wandb.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/wandb.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/__pycache__/wandb.cpython-39.pyc -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/fid.py -------------------------------------------------------------------------------- /utils/pretrained_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/pretrained_resnet.py -------------------------------------------------------------------------------- /utils/sharding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/sharding.py -------------------------------------------------------------------------------- /utils/stable_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/stable_vae.py -------------------------------------------------------------------------------- /utils/train_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/train_state.py -------------------------------------------------------------------------------- /utils/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvfrans/shortcut-models/HEAD/utils/wandb.py --------------------------------------------------------------------------------