Update Dockerfile

Dockerfile

@@ -1,13 +1,5 @@
 FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
 
-ARG OUTPUT_DIR=nroggendorff/zelda-style-diffusion
-ARG INPUT_DS=nroggendorff/zelda-style
-ARG BASE=glides/illustriousxl
-
-ENV OUTPUT_DIR=${OUTPUT_DIR} \
-  INPUT_DS=${INPUT_DS} \
-  BASE=${BASE}
-
 WORKDIR /app
 
 COPY . .
@@ -15,26 +7,34 @@ COPY . .
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
 RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir \
-      "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers" \
-      transformers \
-      accelerate \
-      safetensors \
-      opencv-python \
-      datasets \
-      dataclasses \
-      scikit-learn \
-      huggingface-hub \
-      tensorboard \
-      peft \
-      ftfy \
-      jinja2 \
-      torchvision \
-      torch \
-      --extra-index-url https://download.pytorch.org/whl/cu121
+  pip install --no-cache-dir \
+    "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers" \
+    transformers \
+    accelerate \
+    safetensors \
+    opencv-python \
+    datasets \
+    dataclasses \
+    scikit-learn \
+    huggingface-hub \
+    tensorboard \
+    peft \
+    ftfy \
+    jinja2 \
+    torchvision \
+    torch \
+    --extra-index-url https://download.pytorch.org/whl/cu121
 
 RUN mkdir /.cache && chmod 777 /.cache
 
+ARG OUTPUT_DIR=zelda-style-diffusion
+ARG INPUT_DS=nroggendorff/zelda-style
+ARG BASE=glides/illustriousxl
+
+ENV OUTPUT_DIR=${OUTPUT_DIR} \
+  INPUT_DS=${INPUT_DS} \
+  BASE=${BASE}
+
 RUN mkdir /app/${OUTPUT_DIR} && chmod 777 /app/${OUTPUT_DIR}
 
 RUN accelerate config default