dg845

[WIP] Add UFOGen Pipeline and Scheduler

15

# What does this PR do? This PR adds a pipeline and scheduler for the [UFOGen model](https://arxiv.org/pdf/2311.09257.pdf). The UFOGen model is based on the [denoising diffusion GAN (DDGAN) model](https://arxiv.org/pdf/2112.07804.pdf) with...

LCM Distill Scripts Fix Bug when Initializing Target U-Net

2

# What does this PR do? This PR initializes the `target_unet` target model config from `unet` rather than `teacher_unet` in the LCM full model distillation scripts. This ensures that if...

[WIP] Add Adversarial Diffusion Distillation (ADD) Script

14

# What does this PR do? This PR adds an example script for adversarial diffusion distillation (ADD) ([paper](https://arxiv.org/pdf/2311.17042.pdf), [code](https://github.com/Stability-AI/generative-models)), a distillation + adversarial training method used to distill SD/SD-XL Turbo....

Enable Gradient Checkpointing for UNet2DModel (New)

1

# What does this PR do? This PR supports gradient checkpointing for `UNet2DModel`. Successor to PR #6718. ## Before submitting - [ ] This PR fixes a typo or improves...

[Feature Request] Support `torch.nn.Conv1d` layers for LoRA

2

### Feature request I would like [`torch.nn.Conv1d`](https://pytorch.org/docs/stable/generated/torch.nn.Conv1d.html) layers to be supported by LoRA. Currently, if I try to include a `torch.nn.Conv1d` layer in `LoraConfig.target_modules`, I get the following error: ```bash...