This is the tandem repository to exploit on linux the kohya_ss training webui converted to Linux.
Read the data sections for wheels and packages prior to compiling the image or IT WILL FAIL.
Here is the OFFICIAL Google Collab implementation:
There is a dockerised package of Stable Diffusion AUTOMATIC1111 we maintains here: Stable Diffusion AUTOMATIC1111 P2Enjoy Docker Version.
You may find the profiles to compile tensorflows and xformers there althought there is no (yet) any public instructions on how-to do so: you're free to try to figure it out by yourself or patiently wait until we will find the time to make a guide.
You will need the docker nvidia extensions, please refer to the installations details here: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#setting-up-nvidia-container-toolkit
Remember to allow docker hosts to clients to your X server.
# Somewhat unsecure way to do (I take no responsability if you do this way).
# You should add the exact host but I leave that to you.
xhost local:docker
Once you have built the container, you can pretty just run it via the docker compose --profile kohya up --build
and wait for the build to finish.
A message should notify the build is complete and you can access the gui via the link on the console.
kohya-docker-kohya-1 | Running on local URL: http://127.0.0.1:7680
Happy training on Linux!!!