├── .gitignore ├── LICENSE.txt ├── README.md ├── assets ├── adapters_grid.jpg ├── diagram_full.jpg ├── flash_grid.jpg ├── flash_loras.jpg ├── flash_pixart_grid.jpg ├── flash_sd3.jpg ├── flash_sd_grid.jpg ├── flash_sdxl_grid.jpg ├── inpainting_grid.png ├── swap_grid.png └── upscaler_grid.png ├── examples ├── comfy │ └── comfy_workflow.json ├── configs │ ├── flash_canny_adapter.yaml │ ├── flash_pixart.yaml │ ├── flash_sd.yaml │ ├── flash_sd3.yaml │ └── flash_sdxl.yaml ├── train_flash_canny_adapter.py ├── train_flash_pixart.py ├── train_flash_sd.py ├── train_flash_sd3.py └── train_flash_sdxl.py ├── requirements.txt ├── setup.py ├── src └── flash │ ├── config.py │ ├── data │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ ├── collation_fn.py │ │ ├── dataset.py │ │ └── datasets_config.py │ ├── filters │ │ ├── __init__.py │ │ ├── base.py │ │ ├── filter_wrapper.py │ │ ├── filters.py │ │ └── filters_config.py │ └── mappers │ │ ├── __init__.py │ │ ├── base.py │ │ ├── mappers.py │ │ ├── mappers_config.py │ │ └── mappers_wrapper.py │ ├── models │ ├── __init__.py │ ├── adapters │ │ ├── __init__.py │ │ └── t2i_adapter.py │ ├── base │ │ ├── __init__.py │ │ ├── base_model.py │ │ └── model_config.py │ ├── embedders │ │ ├── __init__.py │ │ ├── base │ │ │ ├── __init__.py │ │ │ ├── base_conditioner.py │ │ │ └── base_conditioner_config.py │ │ ├── clip │ │ │ ├── __init__.py │ │ │ ├── clip_embedder_config.py │ │ │ └── clip_embedder_model.py │ │ ├── conditioners_wrapper.py │ │ ├── t5 │ │ │ ├── __init__.py │ │ │ ├── t5_embedder_config.py │ │ │ └── t5_embedder_model.py │ │ ├── timesteps │ │ │ ├── __init__.py │ │ │ ├── timesteps_embeddding_config.py │ │ │ └── timesteps_embedding.py │ │ └── torch_nn │ │ │ ├── __init__.py │ │ │ ├── embedders.py │ │ │ └── embedders_config.py │ ├── flash │ │ ├── __init__.py │ │ ├── flash_diffusion_config.py │ │ └── flash_diffusion_model.py │ ├── flash_sd3 │ │ ├── __init__.py │ │ ├── flash_diffusion_config.py │ │ └── flash_diffusion_model.py │ ├── transformers │ │ ├── __init__.py │ │ ├── tranformers.py │ │ └── utils.py │ ├── unets │ │ ├── __init__.py │ │ └── unet.py │ ├── utils.py │ └── vae │ │ ├── __init__.py │ │ ├── autoencoderKL.py │ │ └── autoencoderKL_config.py │ └── trainer │ ├── __init__.py │ ├── loggers.py │ ├── trainer.py │ ├── training_config.py │ └── utils.py └── tests ├── README.md ├── requirements.txt ├── test_dataset ├── test_filters.py └── test_mappers.py ├── test_embedders ├── test_clip_embedders.py ├── test_conditioners_wrapper.py ├── test_t5_embedder.py ├── test_time_embedders.py └── test_torchnn_embedder.py ├── test_flash └── test_flash_diffusion.py ├── test_transformers └── test_transformers_wrappers.py ├── test_unet └── test_unets_wrappers.py └── test_vaes └── test_autoencoderKL.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/adapters_grid.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/adapters_grid.jpg -------------------------------------------------------------------------------- /assets/diagram_full.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/diagram_full.jpg -------------------------------------------------------------------------------- /assets/flash_grid.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/flash_grid.jpg -------------------------------------------------------------------------------- /assets/flash_loras.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/flash_loras.jpg -------------------------------------------------------------------------------- /assets/flash_pixart_grid.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/flash_pixart_grid.jpg -------------------------------------------------------------------------------- /assets/flash_sd3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/flash_sd3.jpg -------------------------------------------------------------------------------- /assets/flash_sd_grid.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/flash_sd_grid.jpg -------------------------------------------------------------------------------- /assets/flash_sdxl_grid.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/flash_sdxl_grid.jpg -------------------------------------------------------------------------------- /assets/inpainting_grid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/inpainting_grid.png -------------------------------------------------------------------------------- /assets/swap_grid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/swap_grid.png -------------------------------------------------------------------------------- /assets/upscaler_grid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/assets/upscaler_grid.png -------------------------------------------------------------------------------- /examples/comfy/comfy_workflow.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/comfy/comfy_workflow.json -------------------------------------------------------------------------------- /examples/configs/flash_canny_adapter.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/configs/flash_canny_adapter.yaml -------------------------------------------------------------------------------- /examples/configs/flash_pixart.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/configs/flash_pixart.yaml -------------------------------------------------------------------------------- /examples/configs/flash_sd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/configs/flash_sd.yaml -------------------------------------------------------------------------------- /examples/configs/flash_sd3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/configs/flash_sd3.yaml -------------------------------------------------------------------------------- /examples/configs/flash_sdxl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/configs/flash_sdxl.yaml -------------------------------------------------------------------------------- /examples/train_flash_canny_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/train_flash_canny_adapter.py -------------------------------------------------------------------------------- /examples/train_flash_pixart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/train_flash_pixart.py -------------------------------------------------------------------------------- /examples/train_flash_sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/train_flash_sd.py -------------------------------------------------------------------------------- /examples/train_flash_sd3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/train_flash_sd3.py -------------------------------------------------------------------------------- /examples/train_flash_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/examples/train_flash_sdxl.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/setup.py -------------------------------------------------------------------------------- /src/flash/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/config.py -------------------------------------------------------------------------------- /src/flash/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/__init__.py -------------------------------------------------------------------------------- /src/flash/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/datasets/__init__.py -------------------------------------------------------------------------------- /src/flash/data/datasets/collation_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/datasets/collation_fn.py -------------------------------------------------------------------------------- /src/flash/data/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/datasets/dataset.py -------------------------------------------------------------------------------- /src/flash/data/datasets/datasets_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/datasets/datasets_config.py -------------------------------------------------------------------------------- /src/flash/data/filters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/filters/__init__.py -------------------------------------------------------------------------------- /src/flash/data/filters/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/filters/base.py -------------------------------------------------------------------------------- /src/flash/data/filters/filter_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/filters/filter_wrapper.py -------------------------------------------------------------------------------- /src/flash/data/filters/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/filters/filters.py -------------------------------------------------------------------------------- /src/flash/data/filters/filters_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/filters/filters_config.py -------------------------------------------------------------------------------- /src/flash/data/mappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/mappers/__init__.py -------------------------------------------------------------------------------- /src/flash/data/mappers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/mappers/base.py -------------------------------------------------------------------------------- /src/flash/data/mappers/mappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/mappers/mappers.py -------------------------------------------------------------------------------- /src/flash/data/mappers/mappers_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/mappers/mappers_config.py -------------------------------------------------------------------------------- /src/flash/data/mappers/mappers_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/data/mappers/mappers_wrapper.py -------------------------------------------------------------------------------- /src/flash/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/flash/models/adapters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/adapters/__init__.py -------------------------------------------------------------------------------- /src/flash/models/adapters/t2i_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/adapters/t2i_adapter.py -------------------------------------------------------------------------------- /src/flash/models/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/base/__init__.py -------------------------------------------------------------------------------- /src/flash/models/base/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/base/base_model.py -------------------------------------------------------------------------------- /src/flash/models/base/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/base/model_config.py -------------------------------------------------------------------------------- /src/flash/models/embedders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/__init__.py -------------------------------------------------------------------------------- /src/flash/models/embedders/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/base/__init__.py -------------------------------------------------------------------------------- /src/flash/models/embedders/base/base_conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/base/base_conditioner.py -------------------------------------------------------------------------------- /src/flash/models/embedders/base/base_conditioner_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/base/base_conditioner_config.py -------------------------------------------------------------------------------- /src/flash/models/embedders/clip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/clip/__init__.py -------------------------------------------------------------------------------- /src/flash/models/embedders/clip/clip_embedder_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/clip/clip_embedder_config.py -------------------------------------------------------------------------------- /src/flash/models/embedders/clip/clip_embedder_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/clip/clip_embedder_model.py -------------------------------------------------------------------------------- /src/flash/models/embedders/conditioners_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/conditioners_wrapper.py -------------------------------------------------------------------------------- /src/flash/models/embedders/t5/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/t5/__init__.py -------------------------------------------------------------------------------- /src/flash/models/embedders/t5/t5_embedder_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/t5/t5_embedder_config.py -------------------------------------------------------------------------------- /src/flash/models/embedders/t5/t5_embedder_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/t5/t5_embedder_model.py -------------------------------------------------------------------------------- /src/flash/models/embedders/timesteps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/timesteps/__init__.py -------------------------------------------------------------------------------- /src/flash/models/embedders/timesteps/timesteps_embeddding_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/timesteps/timesteps_embeddding_config.py -------------------------------------------------------------------------------- /src/flash/models/embedders/timesteps/timesteps_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/timesteps/timesteps_embedding.py -------------------------------------------------------------------------------- /src/flash/models/embedders/torch_nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/torch_nn/__init__.py -------------------------------------------------------------------------------- /src/flash/models/embedders/torch_nn/embedders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/torch_nn/embedders.py -------------------------------------------------------------------------------- /src/flash/models/embedders/torch_nn/embedders_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/embedders/torch_nn/embedders_config.py -------------------------------------------------------------------------------- /src/flash/models/flash/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/flash/__init__.py -------------------------------------------------------------------------------- /src/flash/models/flash/flash_diffusion_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/flash/flash_diffusion_config.py -------------------------------------------------------------------------------- /src/flash/models/flash/flash_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/flash/flash_diffusion_model.py -------------------------------------------------------------------------------- /src/flash/models/flash_sd3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/flash_sd3/__init__.py -------------------------------------------------------------------------------- /src/flash/models/flash_sd3/flash_diffusion_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/flash_sd3/flash_diffusion_config.py -------------------------------------------------------------------------------- /src/flash/models/flash_sd3/flash_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/flash_sd3/flash_diffusion_model.py -------------------------------------------------------------------------------- /src/flash/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/transformers/__init__.py -------------------------------------------------------------------------------- /src/flash/models/transformers/tranformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/transformers/tranformers.py -------------------------------------------------------------------------------- /src/flash/models/transformers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/transformers/utils.py -------------------------------------------------------------------------------- /src/flash/models/unets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/unets/__init__.py -------------------------------------------------------------------------------- /src/flash/models/unets/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/unets/unet.py -------------------------------------------------------------------------------- /src/flash/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/utils.py -------------------------------------------------------------------------------- /src/flash/models/vae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/vae/__init__.py -------------------------------------------------------------------------------- /src/flash/models/vae/autoencoderKL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/vae/autoencoderKL.py -------------------------------------------------------------------------------- /src/flash/models/vae/autoencoderKL_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/models/vae/autoencoderKL_config.py -------------------------------------------------------------------------------- /src/flash/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/trainer/__init__.py -------------------------------------------------------------------------------- /src/flash/trainer/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/trainer/loggers.py -------------------------------------------------------------------------------- /src/flash/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/trainer/trainer.py -------------------------------------------------------------------------------- /src/flash/trainer/training_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/trainer/training_config.py -------------------------------------------------------------------------------- /src/flash/trainer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/src/flash/trainer/utils.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/requirements.txt: -------------------------------------------------------------------------------- 1 | pytest 2 | torchmetrics>=1.3.1 -------------------------------------------------------------------------------- /tests/test_dataset/test_filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_dataset/test_filters.py -------------------------------------------------------------------------------- /tests/test_dataset/test_mappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_dataset/test_mappers.py -------------------------------------------------------------------------------- /tests/test_embedders/test_clip_embedders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_embedders/test_clip_embedders.py -------------------------------------------------------------------------------- /tests/test_embedders/test_conditioners_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_embedders/test_conditioners_wrapper.py -------------------------------------------------------------------------------- /tests/test_embedders/test_t5_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_embedders/test_t5_embedder.py -------------------------------------------------------------------------------- /tests/test_embedders/test_time_embedders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_embedders/test_time_embedders.py -------------------------------------------------------------------------------- /tests/test_embedders/test_torchnn_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_embedders/test_torchnn_embedder.py -------------------------------------------------------------------------------- /tests/test_flash/test_flash_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_flash/test_flash_diffusion.py -------------------------------------------------------------------------------- /tests/test_transformers/test_transformers_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_transformers/test_transformers_wrappers.py -------------------------------------------------------------------------------- /tests/test_unet/test_unets_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_unet/test_unets_wrappers.py -------------------------------------------------------------------------------- /tests/test_vaes/test_autoencoderKL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gojasper/flash-diffusion/HEAD/tests/test_vaes/test_autoencoderKL.py --------------------------------------------------------------------------------