update

.dockerignore

@@ -21,4 +21,3 @@ logs/
 outputs/
 sd-scripts/venv
 models/sd_xl_base_1.0.safetensors
-#models/

Dockerfile

@@ -1,5 +1,5 @@
 # Python 3.11 + PyTorch 2.7.0 + CUDA 12.8 + CuDNN 9.5
-FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-devel
+FROM pytorch/pytorch:2.7.1-cuda12.8-cudnn9-devel
 
 # 기본 작업 경로 설정
 WORKDIR /app
@@ -17,16 +17,20 @@ RUN pip install --upgrade pip setuptools wheel
 
 # kohya_ss 전체 복사 (모델 포함)
 COPY . /app/sdxl_train_captioner
-# 두 requirements.txt 모두 설치
+# requirements.txt 설치
 WORKDIR /app/sdxl_train_captioner
-RUN mkdir -p /app/sdxl_train_captioner/models
-RUN mkdir -p /app/sdxl_train_captioner/dataset
 
-RUN pip install -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install flash-attn --no-build-isolation
+
+RUN mkdir -p /app/sdxl_train_captioner/dataset
+RUN mkdir -p /app/sdxl_train_captioner/models
 
 # 모델 파일 복사 (미리 포함시킬 가중치)
 COPY ./models /app/sdxl_train_captioner/models
 
 WORKDIR /app/sdxl_train_captioner/sd-scripts
+
 RUN chmod +x ./entrypoint.sh
+
 ENTRYPOINT ["bash", "entrypoint.sh"]

README.md

@@ -16,7 +16,7 @@ cuDNN (예) C:\Program Files\NVIDIA\CUDNN\v9.5\bin 폴더 안에는 Cuda Major
 C:\Program Files\NVIDIA\CUDNN\v9.5\bin\12.6 아래의 모든 dll 파일을 
 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin으로 복사합니다. 
 
-
+[README.md](../sdxl_train_captioner_runtime/README.md)
 ## 3. SDXL 모델 다운로드
 - 도커 컨테이너가 실행될 때 models 하위에 StableDiffusion XL 1.0 모델이 다운로드 됩니다.
 - 만약에 해당 URL 지원이 종료 된 경우, 허깅페이지 또는 CIVITAI에서 다운로드 하세요.

docker-build.cmd

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

docker-compose.yml

@@ -15,8 +15,8 @@ services:
               count: all
               capabilities: [gpu]
 
-    ports:
-      - "7860:7860"
+    # ports:
+    #   - "7860:7860"
 
     # 볼륨 마운트 (호스트 ↔ 컨테이너)
     volumes:

run-caption-watcher.cmd

@@ -1,6 +0,0 @@
-@echo off
-setx CUDA_VISIBLE_DEVICES "3"
-echo [Watcher] Starting caption watcher...
-python cap-watcher.py --overwrite
-REM --img_dir "../dataset/captioning/mainchar" --out_dir "../dataset/captioning/mainchar"
-pause