├── README.md ├── diffusion_lm ├── README.md ├── catsample.py ├── configs │ ├── config.yaml │ ├── ft.yaml │ └── model │ │ ├── medium.yaml │ │ └── small.yaml ├── eval.py ├── finetune_cond_data.py ├── gfn.py ├── graph_lib.py ├── load_model.py ├── losses.py ├── model │ ├── __init__.py │ ├── ema.py │ ├── fused_add_dropout_scale.py │ ├── rotary.py │ ├── transformer.py │ └── utils.py ├── noise_lib.py ├── requirements.txt ├── results.ipynb ├── run_sample.py ├── run_sample_cond.py ├── run_train.py ├── sampling.py ├── train.py └── utils.py ├── inverse_diffusion ├── README.md ├── requirements.txt ├── samples │ └── cls_finetuning_samples.png └── src │ ├── create_data.py │ ├── finetune_posterior.py │ ├── models │ ├── __init__.py │ ├── classifiers.py │ ├── denoisers.py │ ├── langevin.py │ └── samplers.py │ ├── train_classifier.py │ ├── train_prior.py │ ├── utils │ ├── __init__.py │ ├── args.py │ ├── data_loaders.py │ ├── diffusers │ │ ├── __init__.py │ │ ├── pipelines │ │ │ ├── .DS_Store │ │ │ ├── ddim_gfn │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_ddim_gfn.py │ │ │ ├── ddpm_dp │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_ddpm_dp.py │ │ │ └── ddpm_gfn │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_ddpm.py │ │ └── schedulers │ │ │ ├── scheduling_ddim_gfn.py │ │ │ ├── scheduling_ddpm_dp.py │ │ │ ├── scheduling_ddpm_gfn.py │ │ │ ├── scheduling_edm_euler_gfn.py │ │ │ └── scheduling_sde_ve_gfn.py │ ├── diffusion.py │ ├── fid_evaluation.py │ ├── gfn_diffusion.py │ ├── pytorch_utils.py │ ├── simple_io.py │ └── visualization.py │ └── visualize_runs.py ├── offline_RL ├── IQL_PyTorch │ ├── LICENSE │ ├── README.md │ ├── main.py │ ├── requirements.txt │ ├── results.py │ └── src │ │ ├── iql.py │ │ ├── policy.py │ │ ├── util.py │ │ └── value_functions.py ├── README.md ├── bc_models │ └── .gitkeep ├── model.py ├── q_models │ └── .gitkeep ├── qflow_offline.py └── train_bc.py ├── rtb_diffusion ├── README.md ├── __init__.py ├── classifer_guidance_posterior.py ├── energies │ ├── __init__.py │ ├── base_set.py │ ├── posterior_2Dgmm.py │ └── twenty_five_gmm.py ├── finetune_posterior.py ├── gflownet_losses.py ├── models │ ├── __init__.py │ ├── architectures.py │ └── gfn.py ├── plot_utils.py ├── pretrain_prior.py ├── pretrained │ └── prior.pt └── utils.py └── text_to_image ├── README.md ├── dataset └── drawbench │ └── data_meta.json ├── dpok_utils.py ├── environment.yaml ├── eval_div.py ├── eval_div_clip.py ├── eval_div_dino.py ├── eval_rew.py ├── img └── rabbit.gif ├── install_image_reward.sh ├── pipeline_stable_diffusion_extended.py ├── reward_model.py ├── scheduling_ddim_extended.py ├── test.py ├── train_gfn.py ├── train_online_pg.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/README.md -------------------------------------------------------------------------------- /diffusion_lm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/README.md -------------------------------------------------------------------------------- /diffusion_lm/catsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/catsample.py -------------------------------------------------------------------------------- /diffusion_lm/configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/configs/config.yaml -------------------------------------------------------------------------------- /diffusion_lm/configs/ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/configs/ft.yaml -------------------------------------------------------------------------------- /diffusion_lm/configs/model/medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/configs/model/medium.yaml -------------------------------------------------------------------------------- /diffusion_lm/configs/model/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/configs/model/small.yaml -------------------------------------------------------------------------------- /diffusion_lm/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/eval.py -------------------------------------------------------------------------------- /diffusion_lm/finetune_cond_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/finetune_cond_data.py -------------------------------------------------------------------------------- /diffusion_lm/gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/gfn.py -------------------------------------------------------------------------------- /diffusion_lm/graph_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/graph_lib.py -------------------------------------------------------------------------------- /diffusion_lm/load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/load_model.py -------------------------------------------------------------------------------- /diffusion_lm/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/losses.py -------------------------------------------------------------------------------- /diffusion_lm/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .transformer import SEDD -------------------------------------------------------------------------------- /diffusion_lm/model/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/model/ema.py -------------------------------------------------------------------------------- /diffusion_lm/model/fused_add_dropout_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/model/fused_add_dropout_scale.py -------------------------------------------------------------------------------- /diffusion_lm/model/rotary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/model/rotary.py -------------------------------------------------------------------------------- /diffusion_lm/model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/model/transformer.py -------------------------------------------------------------------------------- /diffusion_lm/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/model/utils.py -------------------------------------------------------------------------------- /diffusion_lm/noise_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/noise_lib.py -------------------------------------------------------------------------------- /diffusion_lm/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/requirements.txt -------------------------------------------------------------------------------- /diffusion_lm/results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/results.ipynb -------------------------------------------------------------------------------- /diffusion_lm/run_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/run_sample.py -------------------------------------------------------------------------------- /diffusion_lm/run_sample_cond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/run_sample_cond.py -------------------------------------------------------------------------------- /diffusion_lm/run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/run_train.py -------------------------------------------------------------------------------- /diffusion_lm/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/sampling.py -------------------------------------------------------------------------------- /diffusion_lm/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/train.py -------------------------------------------------------------------------------- /diffusion_lm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/diffusion_lm/utils.py -------------------------------------------------------------------------------- /inverse_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/README.md -------------------------------------------------------------------------------- /inverse_diffusion/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/requirements.txt -------------------------------------------------------------------------------- /inverse_diffusion/samples/cls_finetuning_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/samples/cls_finetuning_samples.png -------------------------------------------------------------------------------- /inverse_diffusion/src/create_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/create_data.py -------------------------------------------------------------------------------- /inverse_diffusion/src/finetune_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/finetune_posterior.py -------------------------------------------------------------------------------- /inverse_diffusion/src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inverse_diffusion/src/models/classifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/models/classifiers.py -------------------------------------------------------------------------------- /inverse_diffusion/src/models/denoisers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/models/denoisers.py -------------------------------------------------------------------------------- /inverse_diffusion/src/models/langevin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/models/langevin.py -------------------------------------------------------------------------------- /inverse_diffusion/src/models/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/models/samplers.py -------------------------------------------------------------------------------- /inverse_diffusion/src/train_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/train_classifier.py -------------------------------------------------------------------------------- /inverse_diffusion/src/train_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/train_prior.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/args.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/data_loaders.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/.DS_Store -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/ddim_gfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/ddim_gfn/__init__.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/ddim_gfn/pipeline_ddim_gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/ddim_gfn/pipeline_ddim_gfn.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/ddpm_dp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/ddpm_dp/__init__.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/ddpm_dp/pipeline_ddpm_dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/ddpm_dp/pipeline_ddpm_dp.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/ddpm_gfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/ddpm_gfn/__init__.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/pipelines/ddpm_gfn/pipeline_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/pipelines/ddpm_gfn/pipeline_ddpm.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/schedulers/scheduling_ddim_gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/schedulers/scheduling_ddim_gfn.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/schedulers/scheduling_ddpm_dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/schedulers/scheduling_ddpm_dp.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/schedulers/scheduling_ddpm_gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/schedulers/scheduling_ddpm_gfn.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/schedulers/scheduling_edm_euler_gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/schedulers/scheduling_edm_euler_gfn.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusers/schedulers/scheduling_sde_ve_gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusers/schedulers/scheduling_sde_ve_gfn.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/diffusion.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/fid_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/fid_evaluation.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/gfn_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/gfn_diffusion.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/pytorch_utils.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/simple_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/simple_io.py -------------------------------------------------------------------------------- /inverse_diffusion/src/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/utils/visualization.py -------------------------------------------------------------------------------- /inverse_diffusion/src/visualize_runs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/inverse_diffusion/src/visualize_runs.py -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/LICENSE -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/README.md -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/main.py -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/requirements.txt -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/results.py -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/src/iql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/src/iql.py -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/src/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/src/policy.py -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/src/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/src/util.py -------------------------------------------------------------------------------- /offline_RL/IQL_PyTorch/src/value_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/IQL_PyTorch/src/value_functions.py -------------------------------------------------------------------------------- /offline_RL/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/README.md -------------------------------------------------------------------------------- /offline_RL/bc_models/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /offline_RL/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/model.py -------------------------------------------------------------------------------- /offline_RL/q_models/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /offline_RL/qflow_offline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/qflow_offline.py -------------------------------------------------------------------------------- /offline_RL/train_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/offline_RL/train_bc.py -------------------------------------------------------------------------------- /rtb_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/README.md -------------------------------------------------------------------------------- /rtb_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rtb_diffusion/classifer_guidance_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/classifer_guidance_posterior.py -------------------------------------------------------------------------------- /rtb_diffusion/energies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/energies/__init__.py -------------------------------------------------------------------------------- /rtb_diffusion/energies/base_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/energies/base_set.py -------------------------------------------------------------------------------- /rtb_diffusion/energies/posterior_2Dgmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/energies/posterior_2Dgmm.py -------------------------------------------------------------------------------- /rtb_diffusion/energies/twenty_five_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/energies/twenty_five_gmm.py -------------------------------------------------------------------------------- /rtb_diffusion/finetune_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/finetune_posterior.py -------------------------------------------------------------------------------- /rtb_diffusion/gflownet_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/gflownet_losses.py -------------------------------------------------------------------------------- /rtb_diffusion/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .gfn import * 2 | -------------------------------------------------------------------------------- /rtb_diffusion/models/architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/models/architectures.py -------------------------------------------------------------------------------- /rtb_diffusion/models/gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/models/gfn.py -------------------------------------------------------------------------------- /rtb_diffusion/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/plot_utils.py -------------------------------------------------------------------------------- /rtb_diffusion/pretrain_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/pretrain_prior.py -------------------------------------------------------------------------------- /rtb_diffusion/pretrained/prior.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/pretrained/prior.pt -------------------------------------------------------------------------------- /rtb_diffusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/rtb_diffusion/utils.py -------------------------------------------------------------------------------- /text_to_image/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/README.md -------------------------------------------------------------------------------- /text_to_image/dataset/drawbench/data_meta.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/dataset/drawbench/data_meta.json -------------------------------------------------------------------------------- /text_to_image/dpok_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/dpok_utils.py -------------------------------------------------------------------------------- /text_to_image/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/environment.yaml -------------------------------------------------------------------------------- /text_to_image/eval_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/eval_div.py -------------------------------------------------------------------------------- /text_to_image/eval_div_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/eval_div_clip.py -------------------------------------------------------------------------------- /text_to_image/eval_div_dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/eval_div_dino.py -------------------------------------------------------------------------------- /text_to_image/eval_rew.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/eval_rew.py -------------------------------------------------------------------------------- /text_to_image/img/rabbit.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/img/rabbit.gif -------------------------------------------------------------------------------- /text_to_image/install_image_reward.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/install_image_reward.sh -------------------------------------------------------------------------------- /text_to_image/pipeline_stable_diffusion_extended.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/pipeline_stable_diffusion_extended.py -------------------------------------------------------------------------------- /text_to_image/reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/reward_model.py -------------------------------------------------------------------------------- /text_to_image/scheduling_ddim_extended.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/scheduling_ddim_extended.py -------------------------------------------------------------------------------- /text_to_image/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/test.py -------------------------------------------------------------------------------- /text_to_image/train_gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/train_gfn.py -------------------------------------------------------------------------------- /text_to_image/train_online_pg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/train_online_pg.py -------------------------------------------------------------------------------- /text_to_image/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/diffusion-finetuning/HEAD/text_to_image/utils.py --------------------------------------------------------------------------------