search

rinongal / textual_inversion

MIT License
2.9k stars 279 forks source link

How did you prepare the prompts when training the Sstyle? #81

Closed ajundo closed 2 years ago

ajundo commented 2 years ago

Thanks for the great job! I modified the ./ldm/data/personalized.py file, simply replacing all the 'of' by 'in the style of'. I'm not sure if it's correct.

GucciFlipFlops1917 commented 2 years ago

You'll have better luck starting with the style variant of personalized.py (personalized_style.py) and referencing the respective yaml config before running. It's already pre-configured for that purpose.

rinongal commented 2 years ago

As @GucciFlipFlops1917 noted, you can use personalized_style.py. If you're using LDM, you can just use txt2img-1p4B-finetune_style.yaml. Otherwise you can just change the targets in the data block of the yaml to target: ldm.data.personalized_style.PersonalizedBase.