About training my style

[sunmin-lee99]

Thank you for your great work.

I want to train my own image to generate style.
But I feel like the result images are just well generated in certain prompts which are in custom_dataset.py.(ex. A banana on the table)
How can I train my own prompts in styledrop model?

[zideliu]

If you mean sampling results during training, you can edit test_custom_dataset and Discriptor

How can I train my own prompts in styledrop model?

If you want to train your style, modify one_style.json for your style image

[zideliu]

In order to see the training situation well, I fixed the prompt used for sampling during the training process,here

[sunmin-lee99]

Thank you for answering!🙇‍♂️