search

NickLucche / stable-diffusion-nvidia-docker

GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web interface. Includes multi-GPUs support.
MIT License
357 stars 43 forks source link

How to use this docker to train models? #22

Closed winktool closed 1 year ago

winktool commented 1 year ago

Hi.I want to use this docker to train models,what should i do?

NickLucche commented 1 year ago

Hey, this container is for inference only. Please refer to tutorial like https://huggingface.co/docs/diffusers/training/text2image for fine-tuning your own model.