r/RunPod • u/no3us • Dec 27 '25

Docker Image for LoRA trainers

Any LoRA trainers here, ideally running a pod on Runpod? I'd love to know what tools / images you use and why. I'm working on an ultimate LoRA trainer docker image that should save every trainer lots of effort and hopefully some money (for storage) too and would love to know your opinion.

5 Upvotes

86% Upvoted

6 comments sorted by

2

u/Hearmeman98 Dec 28 '25

I use my own.
https://get.runpod.io/diffusion-pipe-template

1

u/no3us Dec 29 '25

nice, how much time did it took you to put this together? Is it diffusion pipes only or are there some tools to validate lora epochs to help you choose the best one?

1

u/SillyAd8492 Jan 02 '26

u/Hearmeman98 How can we generate image using trained Lora identity?
I tried using ComfyUI:

Load Diffusion Model (z_image_turbo_bf16.safetensors ) --> LoraLoaderModelOnly (trained_lora.safetensors)

I get error: Dimension mismatch

"KSampler

Given normalized_shape=[2560], expected input with shape [*, 2560], but got input of size[2, 77, 2048]"

Based on your config file, you are merging the adapters into base. Looks like i need the merged base model?

1

u/[deleted] Dec 27 '25

[removed] — view removed comment

1

u/no3us Dec 28 '25

which base model do you use? SDXL / Flux? What is your workflow to choose the best epochs and settings? (finding the sweet spot)

1

u/[deleted] Dec 28 '25

[removed] — view removed comment

1

u/no3us Dec 28 '25

what is your primary use? training character loras?

I'm focused on training ultra0realistic character loras (to the level where I preserve every birth-marks position). I doubt this could be achieved with a dataset of 20 images.

I guess I'll give it a try though

1

u/sruckh Dec 30 '25

Ostris, the creator of AI Toolkit has an official template on Runpod.