Replicate: All subjects are trained as female

ostris / ai-toolkit

Various AI scripts. Mostly Stable Diffusion stuff.

MIT License

3.49k stars 374 forks source link

Replicate: All subjects are trained as female #111

Open lschaupp opened 3 months ago

lschaupp commented 3 months ago

I am testing the replicate container of this repository. Whenever I am using male subjects as training data, during inference I am obtaining female pictures with their face.

I am using the Auto-Captioning feature (enabled). However even when it is disabled, I am still getting this result.

Anyone had a similar issue?

D-Ogi commented 3 months ago

I would recommend using Florence-2 for image captioning instead.

gadams999 commented 3 months ago

would be good to add Florence-2 as a captioning option. May do a PR for that.