add docker

Author: umag

.github/workflows/publish.yml

@@ -0,0 +1,45 @@
+name: Publish Docker image
+
+on:
+  pull_request:
+    branches: [ sd3 ]
+  workflow_dispatch:
+
+jobs:
+  check_and_build:
+    name: Check for new release and build multiarch Docker image
+    runs-on: ubuntu-latest
+    permissions:
+      packages: write
+      contents: write
+      id-token: write
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v4
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Log in to Docker Hub
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKER_USERNAME }}
+          password: ${{ secrets.DOCKER_PASSWORD }}
+
+      - name: Extract metadata (tags, labels) for Docker
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: umagistr/kohya-sd-scripts
+          tags: |
+            type=raw,value=latest
+
+      - name: Build and push multiarch Docker image
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: Dockerfile
+          push: true
+          platforms: linux/amd64,linux/arm64
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}

Dockerfile

@@ -0,0 +1,36 @@
+# syntax=docker/dockerfile:1
+FROM python:3.10-slim-bullseye
+
+# RUN mount cache for multi-arch
+ARG TARGETARCH
+ARG TARGETVARIANT
+
+# Set environment variables for pip
+ENV PIP_NO_CACHE_DIR=false \
+    PIP_DISABLE_PIP_VERSION_CHECK=1 \
+    PIP_DEFAULT_TIMEOUT=100 \
+    PYTHONUNBUFFERED=1
+
+# Install system dependencies
+RUN --mount=type=cache,id=apt-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/var/cache/apt \
+    --mount=type=cache,id=aptlists-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/var/lib/apt/lists \
+    apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    curl \
+    git \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements file
+COPY requirements.txt .
+
+# Install PyTorch and torchvision with caching
+RUN --mount=type=cache,id=pip-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/pip \
+    pip3 install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu124
+
+# Install Python dependencies with caching
+RUN --mount=type=cache,id=pip-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/pip \
+    pip3 install -r requirements.txt
+
+

requirements.txt

@@ -20,7 +20,6 @@ voluptuous==0.13.1
 huggingface-hub==0.24.5
 # for Image utils
 imagesize==1.4.1
-numpy<=2.0
 # for BLIP captioning
 # requests==2.28.2
 # timm==0.6.12
@@ -43,4 +42,8 @@ rich==13.7.0
 # for T5XXL tokenizer (SD3/FLUX)
 sentencepiece==0.2.0
 # for kohya_ss library
--e .
+#-e .
+torch==2.4.0 
+torchvision==0.19.0
+xformers==0.0.27.post2
+clearml