Not my model!
About this version (zimage turbo v2)
Trained on Diffusion Pipe this time. Up to 30K steps, I have a 50K steps version but I think 30K has least issues. Try it and let me know what you think, if you want a earlier trained or later trained epoch. Recommend strength 0.75 but 0.7-0.85 is the sweet spot. Shift 7.
Changes
-
The penis head shape is better than last version but still not great details on it
-
Going above 1024x1024 can have issues but give it a try and let me know
-
Sitting position is best, standing tends to get hanging penis, crouch doesn't work well.
-
The quality degradation issue is mostly gone
-
Loss of control, some prompts they refuse to grab the penis.
This turbo image model is really frustrating to train. You cannot trouble shoot much because the seeds are so fixed. As a result, some prompts are just cursed and will never work. Qwen picks up the penis shape very quick with little issue. I uploaded 50K steps 489 epoch if you wanna try in the "training data" as a zip file.
I do not think I will do another attempt, we better wait for the base model to train it properly.
Trigger word:
PENISLORA
What can this lora do?
This lora can add erect penises to both men or women viewed from the front/side. Other angles such as POV may have a backwards penis head.
Other things it can now do:
Side view of the penis
Cumming / Cumshots
Blowjobs (its captioned for the words "blowjob" and "deepthroat" )
What can't it do?
No penetration in the training data. Also nothing from POV angle, though there is a few images from above and 1 POV video in the training data.
Sometimes blowjobs with cumming have the penis slip out the closed mouth.
Recommended Settings
It works pretty good with the new lightning dyno high model. I'll link to it in my example workflow. I like to use dyno high model (no lightning lora), then for low I use the lightning v2 lora on the regular 2.2 low base model.
Dataset
84 images at 512x resolution
43 videos at 256x resolution
(I let DP pick the aspect ratio automatically)
This is the same exact dataset as the 2.2 5B model. I made no changes.
Training
I used the default diffusion pipe settings.
[optimizer]
type = 'adamw_optimi'
lr = 2e-5
betas = [0.9, 0.99]
weight_decay = 0.01
eps = 1e-8
I was baffled why it was taking so long to train the high until I realized after over 60 hours of training that I had put my videos in the images directory which resulted in the high being trained ONLY only on videos and twice (once with a very high resolution). Once I fixed this, I went back and trained from 11K steps up to around 13K with the images in the training data. The high model was fine without to be honest.
For the low, I trained it properly with videos and images the whole way, around 6K steps in I upped the image resolution from 512 to 1024 actually and didn't get an OOM (it fit around 24GB exactly). I trained it to around 10.5K steps. Also I trained the low on the full timestep range (0 to 1 instead of 0 to 0.85) from some advice, it may switch better over from high to low on the speed up lora with low steps.
I think I might do another version with more angles such as POV and from the behind to make this work for any situation. In that case I don't think it needs 10K steps per training session, epochs around 5K steps looked fine.
The results
I think it was a combination of improved captioning and 2.2 base model being better. But this lora turned out really well.