Nerogar / OneTrainer

OneTrainer is a one-stop solution for all your stable diffusion training needs.
GNU Affero General Public License v3.0
1.67k stars 136 forks source link

[Feat]: Add seed and prompt to sample images #306

Open RandomTurtle opened 4 months ago

RandomTurtle commented 4 months ago

Describe your use-case.

When I use Automatic1111 to check the PNG info of an image generated using OneTrainer it says there is no prompt. There are great images I would like to re-generate using a different sampler but I don't have any way to find the seed and prompt.

What would you like to see as a solution?

*

Have you considered alternatives? List them here.

An alternative to attaching the seed and prompt to the image would be to output the seed and prompt to the command window, or a text file that (optionally) accompanies the image, if embedding into the image would be a lot of work.

Nerogar commented 4 months ago

None of the settings of OneTrainer are compatible with A1111. Even if the seed, sampler, cfg etc. Are saved in the metadata, using the same settings in A1111 will create different results.

RandomTurtle commented 4 months ago

I mean I used Automatic1111 only to pull for the prompt and seed, so I could generate in Onetrainer.

O-J1 commented 2 months ago

What you are asking for is for prompt metadata to be embeded into the samples just like how A111 does it? Is this correct?