Skip to content

Commit a757b2d

Browse files
if dreambooth lora (#3360)
* update IF stage I pipelines add fixed variance schedulers and lora loading * added kv lora attn processor * allow loading into alternative lora attn processor * make vae optional * throw away predicted variance * allow loading into added kv lora layer * allow load T5 * allow pre compute text embeddings * set new variance type in schedulers * fix copies * refactor all prompt embedding code class prompts are now included in pre-encoding code max tokenizer length is now configurable embedding attention mask is now configurable * fix for when variance type is not defined on scheduler * do not pre compute validation prompt if not present * add example test for if lora dreambooth * add check for train text encoder and pre compute text embeddings
1 parent 571bc1e commit a757b2d

File tree

7 files changed

+382
-44
lines changed

7 files changed

+382
-44
lines changed

0 commit comments

Comments
 (0)