[WIP] Add Unet for consistency models#1

Merged

dg845 merged 34 commits intodg845:consistency-models-pipelinefrom

ayushtues:consistency_unet

May 25, 2023

Commits on May 16, 2023

Commits on May 17, 2023

Allow arbitrary aspect ratio in IFSuperResolutionPipeline (#3298)

devxpy
and
patrickvonplaten
authored
Adding 'strength' parameter to StableDiffusionInpaintingPipeline (#3424)

authored
[WIP] Bugfix - Pipeline.from_pretrained is broken when the pipeline is partially downloaded (#3448)
vimarshc
authored
Fix gradient checkpointing bugs in freezing part of models (requires_grad=False) (#3404)

IrisRainbowNeko
and
patrickvonplaten
authored
Make dreambooth lora more robust to orig unet (#3462)
patrickvonplaten
authored
Reduce peak VRAM by releasing large attention tensors (as soon as they're unnecessary) (#3463)
cmdr2
authored
Add min snr to text2img lora training script (#3459)
wfng92
authored
Add inpaint lora scale support (#3460)
Glaceon-Hyy
authored
[From ckpt] Fix from_ckpt (#3466)
patrickvonplaten
authored
Update full dreambooth script to work with IF (#3425)
williamberman
authored
Add IF dreambooth docs (#3470)
williamberman
authored

Commits on May 18, 2023

Commits on May 20, 2023

mps & onnx tests rework (#3449)
pcuenca
authored

Commits on May 21, 2023

Commits on May 22, 2023

Merge branch 'main2' into consistency_unet
ayushmangal
committed
Add Unet blocks for consistency models
ayushmangal
committed
Add conversion script for Unet
ayushmangal
committed
Fix bug in new unet blocks
ayushmangal
committed

Commits on May 23, 2023

Fix attention weight loading
ayushmangal
committed