update

.gitignore

@@ -59,6 +59,6 @@ sd-scripts/**
 venv
 venv*
 .python-version
-output_model
-output_model/
-output_model/**
+output_models
+output_models/
+output_models/**

Dockerfile

@@ -1,5 +1,5 @@
 # CUDA 12.1 + PyTorch 2.3.0
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-devel
+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
 
 # 기본 작업 경로 설정
 WORKDIR /app
@@ -19,6 +19,13 @@ RUN pip install --upgrade pip setuptools wheel
 COPY . /app/sdxl_train_captioner
 # 두 requirements.txt 모두 설치
 WORKDIR /app/sdxl_train_captioner
+RUN mkdir /app/sdxl_train_captioner/logs
+RUN mkdir /app/sdxl_train_captioner/output_models
+RUN mkdir /app/sdxl_train_captioner/captioning/background
+RUN mkdir /app/sdxl_train_captioner/captioning/mainchar
+RUN mkdir /app/sdxl_train_captioner/training/background
+RUN mkdir /app/sdxl_train_captioner/training/mainchar
+
 RUN pip install -r requirements.txt
 
 # 모델 파일 복사 (미리 포함시킬 가중치)
@@ -27,7 +34,8 @@ COPY ./models /app/sdxl_train_captioner/models
 # 모델 디렉토리 확인 로그
 RUN echo "✅ Copied models:" && ls -R /app/sdxl_train_captioner/models || echo "⚠️ No models found"
 
-RUN chmod +x entrypoint.sh
+WORKDIR /app/sdxl_train_captioner/sd-scripts
+RUN chmod +x ./entrypoint.sh
 ENTRYPOINT ["bash", "entrypoint.sh"]
 
 # 엔트리포인트 복사 및 실행 권한

config_files/config-5080.json

@@ -25,7 +25,7 @@
       "save_every_n_steps": 1000
     },
     "folders": {
-      "train_data_dir": "./data/train",
+      "train_data_dir": "./data/training",
       "reg_data_dir": "./data/reg",
       "output_dir": "./output_5080",
       "logging_dir": "./logs_5080"

config_files/config-5090.json

@@ -25,7 +25,7 @@
     "save_every_n_steps": 1000
   },
   "folders": {
-    "train_data_dir": "./data/train",
+    "train_data_dir": "./data/training",
     "reg_data_dir": "./data/reg",
     "output_dir": "./output_5090",
     "logging_dir": "./logs_5090"

docker-build.cmd

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

docker-compose.yml

@@ -22,7 +22,7 @@ services:
     volumes:
       - ./models:/app/sdxl_train_captioner/models
       - ./dataset:/app/sdxl_train_captioner/dataset
-      - ./outputs:/app/sdxl_train_captioner/output_model
+      - ./outputs:/app/sdxl_train_captioner/output_models
 
     environment:
       # GPU 선택 (필요 시 GPU ID 지정)

docker-down.cmd

@@ -0,0 +1 @@
+docker-compose down --volumes --remove-orphans sdxl_train_captioner

docker-start.cmd

@@ -0,0 +1 @@
+docker-compose start sdxl_train_captioner

docker-stop.cmd

@@ -0,0 +1 @@
+docker-compose stop sdxl_train_captioner

sd-scripts

@@ -1 +1 @@
-Subproject commit 23c37369c9a5c74120c9a11efc9d7772d343be35
+Subproject commit 2431c4673602cd7e42c6cdcafec951d811366505