Skip to content

Commit

Permalink
Stable diffusion text2img conversion script. (huggingface#154)
Browse files Browse the repository at this point in the history
* begin text2img conversion script

* add fn to convert config

* create config if not provided

* update imports and use UNet2DConditionModel

* fix imports, layer names

* fix unet coversion

* add function to convert VAE

* fix vae conversion

* update main

* create text model

* update config creating logic for unet

* fix config creation

* update script to create and save pipeline

* remove unused imports

* fix checkpoint loading

* better name

* save progress

* finish

* up

* up

Co-authored-by: Patrick von Platen <[email protected]>
  • Loading branch information
patil-suraj and patrickvonplaten authored Sep 15, 2022
1 parent d8b0e4f commit 039958e
Showing 1 changed file with 690 additions and 0 deletions.
Loading

0 comments on commit 039958e

Please sign in to comment.