https://github.com/ai-dock/kohya_ss
Kohya's GUI docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.
https://github.com/ai-dock/kohya_ss
artificial-intelligence cloud docker machine-learning python stable-diffusion vast-ai
Last synced: about 1 year ago
JSON representation
Kohya's GUI docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.
- Host: GitHub
- URL: https://github.com/ai-dock/kohya_ss
- Owner: ai-dock
- License: other
- Created: 2024-05-31T14:20:28.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-11-15T18:14:26.000Z (over 1 year ago)
- Last Synced: 2025-05-08T03:54:39.238Z (about 1 year ago)
- Topics: artificial-intelligence, cloud, docker, machine-learning, python, stable-diffusion, vast-ai
- Language: Shell
- Homepage:
- Size: 45.9 KB
- Stars: 14
- Watchers: 1
- Forks: 8
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE.md
Awesome Lists containing this project
README
[](https://github.com/ai-dock/kohya_ss/actions/workflows/docker-build.yml)
# Kohya's GUI (kohya_ss) Docker Image
Run [Kohya's GUI](https://github.com/bmaltais/kohya_ss) in a docker container locally or in the cloud.
>[!NOTE]
>These images do not bundle models or third-party configurations. You should use a [provisioning script](https://github.com/ai-dock/base-image/wiki/4.0-Running-the-Image#provisioning-script) to automatically configure your container. You can find examples in `config/provisioning`.
## Documentation
All AI-Dock containers share a common base which is designed to make running on cloud services such as [vast.ai](https://link.ai-dock.org/vast.ai) as straightforward and user friendly as possible.
Common features and options are documented in the [base wiki](https://github.com/ai-dock/base-image/wiki) but any additional features unique to this image will be detailed below.
#### Version Tags
The `:latest` tag points to `:latest-cuda`
Tags follow these patterns:
##### _CUDA_
- `:v2-cuda-[x.x.x]-base-[ubuntu-version]`
- `:latest-cuda` → `:v2-cuda-12.1.1-base-22.04`
##### _ROCm_
- `:v2-rocm-[x.x.x]-core-[ubuntu-version]`
- `:latest-rocm` → `:v2-rocm-6.0-core-22.04`
Browse [here](https://github.com/ai-dock/kohya_ss/pkgs/container/kohya_ss) for an image suitable for your target environment.
Supported Python versions: `3.10`
Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`
## Additional Environment Variables
| Variable | Description |
| ------------------------ | ----------- |
| `AUTO_UPDATE` | Update Kohya_ss on startup (default `false`) |
| `KOHYA_ARGS` | Startup arguments |
| `KOHYA_PORT_HOST` | Kohya's GUI port (default `7860`) |
| `KOHYA_REF` | Git reference for auto update. Accepts branch, tag or commit hash. Default: latest release |
| `KOHYA_URL` | Override `$DIRECT_ADDRESS:port` with URL for Kohya's GUI |
| `TENSORBOARD_ARGS` | Startup arguments (default `--logdir /opt/kohya_ss/logs`) |
| `TENSORBOARD_PORT_HOST` | Tensorboard port (default `6006`) |
| `TENSORBOARD_URL` | Override `$DIRECT_ADDRESS:port` with URL for Tensorboard |
See the base environment variables [here](https://github.com/ai-dock/base-image/wiki/2.0-Environment-Variables) for more configuration options.
### Additional Python Environments
| Environment | Packages |
| -------------- | ----------------------------------------- |
| `kohya` | Kohya's GUI and dependencies |
This virtualenv will be activated on shell login.
~~See the base environments [here](https://github.com/ai-dock/base-image/wiki/1.0-Included-Software#installed-micromamba-environments).~~
## Additional Services
The following services will be launched alongside the [default services](https://github.com/ai-dock/base-image/wiki/1.0-Included-Software) provided by the base image.
### Kohya's GUI
The service will launch on port `7860` unless you have specified an override with `KOHYA_PORT`.
You can set startup arguments by using variable `KOHYA_ARGS`.
To manage this service you can use `supervisorctl [start|stop|restart] kohya_ss`.
### Tensorboard
The service will launch on port `6006` unless you have specified an override with `TENSORBOARD_PORT`.
To manage this service you can use `supervisorctl [start|stop|restart] tensorboard`.
>[!NOTE]
>All services are password protected by default. See the [security](https://github.com/ai-dock/base-image/wiki#security) and [environment variables](https://github.com/ai-dock/base-image/wiki/2.0-Environment-Variables) documentation for more information.
## Pre-Configured Templates
**Vast.​ai**
- [Kohya's GUI:latest-cuda](https://link.ai-dock.org/template-vast-kohya_ss)
- [Kohya's GUI:latest-rocm](https://link.ai-dock.org/template-vast-kohya_ss-rocm)
---
_The author ([@robballantyne](https://github.com/robballantyne)) may be compensated if you sign up to services linked in this document. Testing multiple variants of GPU images in many different environments is both costly and time-consuming; This helps to offset costs_