├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── docs └── source │ └── imgs │ └── diffusers_library.jpg ├── examples ├── README.md ├── experimental │ ├── train_glide_text_to_image.py │ └── train_latent_text_to_image.py └── train_unconditional.py ├── pyproject.toml ├── run.py ├── scripts ├── conversion_bddm.py ├── conversion_glide.py └── conversion_ldm_uncond.py ├── setup.cfg ├── setup.py ├── src └── diffusers │ ├── __init__.py │ ├── configuration_utils.py │ ├── dependency_versions_check.py │ ├── dependency_versions_table.py │ ├── dynamic_modules_utils.py │ ├── hub_utils.py │ ├── modeling_utils.py │ ├── models │ ├── README.md │ ├── __init__.py │ ├── attention.py │ ├── embeddings.py │ ├── resnet.py │ ├── unet.py │ ├── unet_glide.py │ ├── unet_grad_tts.py │ ├── unet_ldm.py │ ├── unet_new.py │ ├── unet_rl.py │ ├── unet_sde_score_estimation.py │ ├── unet_unconditional.py │ └── vae.py │ ├── optimization.py │ ├── pipeline_utils.py │ ├── pipelines │ ├── README.md │ ├── __init__.py │ ├── bddm │ │ ├── __init__.py │ │ └── pipeline_bddm.py │ ├── ddim │ │ ├── __init__.py │ │ └── pipeline_ddim.py │ ├── ddpm │ │ ├── __init__.py │ │ └── pipeline_ddpm.py │ ├── glide │ │ ├── __init__.py │ │ └── pipeline_glide.py │ ├── grad_tts │ │ ├── __init__.py │ │ ├── grad_tts_utils.py │ │ └── pipeline_grad_tts.py │ ├── latent_diffusion │ │ ├── __init__.py │ │ └── pipeline_latent_diffusion.py │ ├── latent_diffusion_uncond │ │ ├── __init__.py │ │ └── pipeline_latent_diffusion_uncond.py │ ├── pndm │ │ ├── __init__.py │ │ └── pipeline_pndm.py │ ├── score_sde_ve │ │ ├── __init__.py │ │ └── pipeline_score_sde_ve.py │ └── score_sde_vp │ │ ├── __init__.py │ │ └── pipeline_score_sde_vp.py │ ├── schedulers │ ├── README.md │ ├── __init__.py │ ├── scheduling_ddim.py │ ├── scheduling_ddpm.py │ ├── scheduling_grad_tts.py │ ├── scheduling_pndm.py │ ├── scheduling_sde_ve.py │ ├── scheduling_sde_vp.py │ └── scheduling_utils.py │ ├── testing_utils.py │ ├── training_utils.py │ └── utils │ ├── __init__.py │ ├── dummy_transformers_and_inflect_and_unidecode_objects.py │ ├── dummy_transformers_objects.py │ ├── logging.py │ └── model_card_template.md ├── tests ├── __init__.py ├── test_layers_utils.py ├── test_modeling_utils.py └── test_scheduler.py └── utils ├── check_config_docstrings.py ├── check_copies.py ├── check_dummies.py ├── check_inits.py ├── check_repo.py ├── check_table.py ├── check_tf_ops.py └── custom_init_isort.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/README.md -------------------------------------------------------------------------------- /docs/source/imgs/diffusers_library.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/docs/source/imgs/diffusers_library.jpg -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/experimental/train_glide_text_to_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/examples/experimental/train_glide_text_to_image.py -------------------------------------------------------------------------------- /examples/experimental/train_latent_text_to_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/examples/experimental/train_latent_text_to_image.py -------------------------------------------------------------------------------- /examples/train_unconditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/examples/train_unconditional.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- 1 | [tool.black] 2 | line-length = 119 3 | target-version = ['py36'] 4 | -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/run.py -------------------------------------------------------------------------------- /scripts/conversion_bddm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/scripts/conversion_bddm.py -------------------------------------------------------------------------------- /scripts/conversion_glide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/scripts/conversion_glide.py -------------------------------------------------------------------------------- /scripts/conversion_ldm_uncond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/scripts/conversion_ldm_uncond.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/setup.py -------------------------------------------------------------------------------- /src/diffusers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/__init__.py -------------------------------------------------------------------------------- /src/diffusers/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/configuration_utils.py -------------------------------------------------------------------------------- /src/diffusers/dependency_versions_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/dependency_versions_check.py -------------------------------------------------------------------------------- /src/diffusers/dependency_versions_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/dependency_versions_table.py -------------------------------------------------------------------------------- /src/diffusers/dynamic_modules_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/dynamic_modules_utils.py -------------------------------------------------------------------------------- /src/diffusers/hub_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/hub_utils.py -------------------------------------------------------------------------------- /src/diffusers/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/modeling_utils.py -------------------------------------------------------------------------------- /src/diffusers/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/README.md -------------------------------------------------------------------------------- /src/diffusers/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/__init__.py -------------------------------------------------------------------------------- /src/diffusers/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/attention.py -------------------------------------------------------------------------------- /src/diffusers/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/embeddings.py -------------------------------------------------------------------------------- /src/diffusers/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/resnet.py -------------------------------------------------------------------------------- /src/diffusers/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_glide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_glide.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_grad_tts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_grad_tts.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_ldm.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_new.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_rl.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_sde_score_estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_sde_score_estimation.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_unconditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/unet_unconditional.py -------------------------------------------------------------------------------- /src/diffusers/models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/models/vae.py -------------------------------------------------------------------------------- /src/diffusers/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/optimization.py -------------------------------------------------------------------------------- /src/diffusers/pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipeline_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/README.md -------------------------------------------------------------------------------- /src/diffusers/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/bddm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/bddm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/bddm/pipeline_bddm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/bddm/pipeline_bddm.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/ddim/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddim/pipeline_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/ddim/pipeline_ddim.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddpm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/ddpm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddpm/pipeline_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/ddpm/pipeline_ddpm.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/glide/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/glide/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/glide/pipeline_glide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/glide/pipeline_glide.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/grad_tts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/grad_tts/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/grad_tts/grad_tts_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/grad_tts/grad_tts_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/grad_tts/pipeline_grad_tts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/grad_tts/pipeline_grad_tts.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/latent_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/latent_diffusion/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/latent_diffusion_uncond/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/latent_diffusion_uncond/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pndm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/pndm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pndm/pipeline_pndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/pndm/pipeline_pndm.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/score_sde_ve/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/score_sde_ve/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/score_sde_vp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/score_sde_vp/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/score_sde_vp/pipeline_score_sde_vp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/pipelines/score_sde_vp/pipeline_score_sde_vp.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/README.md -------------------------------------------------------------------------------- /src/diffusers/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/__init__.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_ddim.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_ddpm.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_grad_tts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_grad_tts.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_pndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_pndm.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_sde_ve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_sde_ve.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_sde_vp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_sde_vp.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/schedulers/scheduling_utils.py -------------------------------------------------------------------------------- /src/diffusers/testing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/testing_utils.py -------------------------------------------------------------------------------- /src/diffusers/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/training_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/utils/__init__.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_transformers_and_inflect_and_unidecode_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/utils/dummy_transformers_and_inflect_and_unidecode_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_transformers_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/utils/dummy_transformers_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/utils/logging.py -------------------------------------------------------------------------------- /src/diffusers/utils/model_card_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/src/diffusers/utils/model_card_template.md -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_layers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/tests/test_layers_utils.py -------------------------------------------------------------------------------- /tests/test_modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/tests/test_modeling_utils.py -------------------------------------------------------------------------------- /tests/test_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/tests/test_scheduler.py -------------------------------------------------------------------------------- /utils/check_config_docstrings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_config_docstrings.py -------------------------------------------------------------------------------- /utils/check_copies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_copies.py -------------------------------------------------------------------------------- /utils/check_dummies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_dummies.py -------------------------------------------------------------------------------- /utils/check_inits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_inits.py -------------------------------------------------------------------------------- /utils/check_repo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_repo.py -------------------------------------------------------------------------------- /utils/check_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_table.py -------------------------------------------------------------------------------- /utils/check_tf_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/check_tf_ops.py -------------------------------------------------------------------------------- /utils/custom_init_isort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/diffusers_all/HEAD/utils/custom_init_isort.py --------------------------------------------------------------------------------