Commit Graph

2 Commits

Author SHA1 Message Date
Hayk Martiros d820e1fecf Fix minor errors
Topic: clean_rewrite
2022-12-26 17:47:11 -08:00
Hayk Martiros 2fb1153ec8 Refactor RiffusionPipeline
* Bunch of cleanup and typing
 * Move prompt_weighting to be marked as external
 * Add helpers for loading the checkpoint and traced unet

Topic: clean_rewrite
2022-12-26 17:47:11 -08:00