How can I fine tune the model with autotrain?

#6
by mahammady - opened

Hello. I would love to fine tune this model and generate my own images. How can I do that?
I have used autotrain on stable diffusion base xl model. But using it on your model seems not quite right.

autotrain dreambooth --model SG161222/Realistic_Vision_V6.0_B1_noVAE --image-path input_images/ --prompt "photo of maha person" --resolution 1024 --batch-size 1 --num-steps 500 --mixed-precision fp16 --gradient-accumulation 4 --lr 1e-4 --project-name realistic-vision

this gives me pytorch_lora_weights.safetensors file and I use this to generate my own images.
pipe.load_lora_weights("model/", weight_name="pytorch_lora_weights.safetensors")

Is this correct technique of fine-tuning Realistic_Vision_V6.0_B1_noVAE model?

Sign up or log in to comment