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-25 19:34:24 +09:00
parent 98f7db33dc
commit 431e9fa4ff
4 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -6,7 +6,7 @@ WORKDIR /app
# 필수 패키지 설치
RUN apt-get update && apt-get install -y --no-install-recommends \
git wget curl libgl1 libglib2.0-0 \
git wget curl libgl1 libglib2.0-0 libcudnn9 libcudnn9-dev \
&& rm -rf /var/lib/apt/lists/*
# Python 패키지 캐싱 방지

4
docker-build.cmd
View File

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

1
docker-push.cmd Normal file
View File

@ -0,0 +1 @@
docker push aicompanion/sdxl_train_captioner:latest

8
entrypoint.sh
View File

@ -1,9 +1,5 @@
#!/bin/bash
# 모델 다운로드
if [ ! -f /app/models/sd_xl_base_1.0.safetensors ]; then
wget -O /app/models/sd_xl_base_1.0.safetensors https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors
if [ ! -f /app/sdxl_train_captioner/models/sd_xl_base_1.0.safetensors ]; then
wget -O /app/sdxl_train_captioner/models/sd_xl_base_1.0.safetensors https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors
fi
# 캡셔너 실행
#cd sd-scripts
#python cap-watcher.py