1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update

pull/3451/head
sungjoonkim 2025-10-26 01:38:54 +09:00
parent 799d0afb3b
commit f8198ab7a9
5 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -1,4 +1,4 @@
# CUDA 12.1 + PyTorch 2.3.0
# Python 3.11 + PyTorch 2.4.0 + CUDA 12.1 + CuDNN 9.5
FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
# 기본 작업 경로 설정

4
docker-build.cmd
View File

@ -1,3 +1,3 @@
docker build --no-cache -t aicompanion/sdxl_train_captioner:0.8.5 .
docker build --no-cache -t aicompanion/sdxl_train_captioner:0.8.6 .
docker tag aicompanion/sdxl_train_captioner:0.8.5 aicompanion/sdxl_train_captioner:latest
docker tag aicompanion/sdxl_train_captioner:0.8.6 aicompanion/sdxl_train_captioner:latest

2
docker-compose.yml
View File

@ -22,7 +22,7 @@ services:
volumes:
- ./models:/app/sdxl_train_captioner/models
- ./dataset:/app/sdxl_train_captioner/dataset
- ./outputs:/app/sdxl_train_captioner/output_models
- ./output_models:/app/sdxl_train_captioner/output_models
- ./logs:/app/sdxl_train_captioner/logs
environment:

2
requirements.txt
View File

@ -47,6 +47,6 @@ open-clip-torch==2.20.0
rich==13.7.0
# Kohya_ss library
xformers==0.0.26.post1 --index-url https://download.pytorch.org/whl/cu121
xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu121
triton==2.3.0

2
sd-scripts

@ -1 +1 @@
Subproject commit cbe231a17a257baa9c716b56208e891508cfb6b6
Subproject commit 1614bc4863161f270a89e466cd30aa8891192447