Files

k_diffusion

Add modified DPM++ 2M sampler by @hallatore

Apr 10, 2023

c41588a · Apr 10, 2023

This branch is 9 commits ahead of, 11 commits behind Birch-san/k-diffusion:mps.

Name		Name	Last commit message	Last commit date
parent directory ..
models		models	Use orthogonal init for 1x1s that change channel count in ResConvBloc…	Sep 12, 2022
__init__.py		__init__.py	Add v-diffusion-pytorch and OpenAI model wrappers	Jul 17, 2022
augmentation.py		augmentation.py	Add progressive growing	Jul 17, 2022
config.py		config.py	Provide default values for the log logistic sample density	Sep 25, 2022
evaluation.py		evaluation.py	Disable TF32 for FID and KID calculation	Sep 11, 2022
external.py		external.py	Add device and version checks	Apr 10, 2023
gns.py		gns.py	Improve the DDP gradient stats hook's API	Sep 9, 2022
layers.py		layers.py	Fix layers.UnconditionedModule	Sep 16, 2022
sampling.py		sampling.py	Add modified DPM++ 2M sampler by @hallatore	Apr 10, 2023
utils.py		utils.py	Add device and version checks	Apr 10, 2023