Skip to content

Commit c221086

Browse files
williambermandg845
authored andcommitted
if dreambooth lora (huggingface#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 6644d11 commit c221086

File tree

7 files changed

+382
-44
lines changed

7 files changed

+382
-44
lines changed

0 commit comments

Comments
 (0)