From 72ac7314592548aabbe5b0db8dc1702e7d192bf2 Mon Sep 17 00:00:00 2001
From: "jakub.binkowski" <jakub.binkowski@pwr.edu.pl>
Date: Fri, 13 Dec 2024 06:26:59 +0000
Subject: [PATCH] Refine sif building

---
 slurm/build_apptainer.sh         | 9 ++++++---
 slurm/fine_tuning_env.dockerfile | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/slurm/build_apptainer.sh b/slurm/build_apptainer.sh
index 3bd2518..1f40b69 100755
--- a/slurm/build_apptainer.sh
+++ b/slurm/build_apptainer.sh
@@ -6,7 +6,7 @@ set -e
 
 # Creates docker image containing the environment for fine-tuning.
 docker build \
-    --tag juddges_sft:latest \
+    --tag juddges:latest \
     --file ./slurm/fine_tuning_env.dockerfile \
     .
 
@@ -14,5 +14,8 @@ docker build \
 docker run \
     --rm \
     --volume /var/run/docker.sock:/var/run/docker.sock \
-    --volume $(pwd):/juddges \
-    kaczmarj/apptainer build juddges_sft.sif docker-daemon://juddges_sft:latest
+    --volume $(pwd):/work \
+    kaczmarj/apptainer \
+        build \
+        juddges.sif \
+        docker-daemon://juddges:latest
diff --git a/slurm/fine_tuning_env.dockerfile b/slurm/fine_tuning_env.dockerfile
index 4bd8f22..f5a4e44 100644
--- a/slurm/fine_tuning_env.dockerfile
+++ b/slurm/fine_tuning_env.dockerfile
@@ -1,6 +1,6 @@
 FROM nvcr.io/nvidia/pytorch:24.11-py3
 
-WORKDIR /judddges
+WORKDIR /juddges
 
 RUN apt-get update -qq && apt-get install --yes -q make git