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 19:48:05 +09:00
parent f888c49561
commit 21808c5830
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@ -1,5 +1,5 @@
# Python 3.11 + PyTorch 2.4.0 + CUDA 12.1 + CuDNN 9.5
FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
# Python 3.11 + PyTorch 2.7.0 + CUDA 12.8 + CuDNN 9.5
FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-devel
# 기본 작업 경로 설정
WORKDIR /app

4
docker-build.cmd
View File

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

4
requirements.txt
View File

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