search

cloneofsimo / lora

Using Low-rank adaptation to quickly fine-tune diffusion models.
https://arxiv.org/abs/2106.09685
Apache License 2.0
6.94k stars 479 forks source link

Support captions list in preprocess files #193

Closed A2va closed 1 year ago

A2va commented 1 year ago

With this change, it's possible to preprocess files with manual captioning

A2va commented 1 year ago

Also I have a question, in cli_lora_pti.py we can add more than one concept (person/style/object) but how to train two concepts with the same data set ? Do I need to caption files with the concept token ?

cloneofsimo commented 1 year ago

Thanks! You can optionally provide captions and use token_map to train two concepts