new image?

Author: janpf

.github/workflows/dockerimage.yml

@@ -5,7 +5,7 @@ name: Create and publish a Docker image
 on:
   push:
     branches:
-      - main
+      - germeval25
 
 # Defines two custom environment variables for the workflow. These are used for the Container registry domain, and a name for the Docker image that this workflow builds.
 env:
@@ -50,7 +50,7 @@ jobs:
         uses: docker/metadata-action@9ec57ed1fcdbf14dcef7dfbe97b2010124a938b7
         with:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
-          tags: main
+          tags: germeval25
       # This step uses the `docker/build-push-action` action to build the image, based on your repository's `Dockerfile`. If the build succeeds, it pushes the image to GitHub Packages.
       # It uses the `context` parameter to define the build's context as the set of files located in the specified path. For more information, see "[Usage](https://github.com/docker/build-push-action#usage)" in the README of the `docker/build-push-action` repository.
       # It uses the `tags` and `labels` parameters to tag and label the image with the output from the "meta" step.

k8s/templates/Dockerfile

@@ -33,8 +33,8 @@ RUN chown -R ${USER}:${USER} /localdir/
 
 COPY . .
 RUN python -m pip install packaging wheel
-RUN python -m pip install --extra-index-url https://download.pytorch.org/whl/cu121 torch==2.3.1+cu121
-RUN python -m pip install --no-build-isolation --extra-index-url https://download.pytorch.org/whl/cu121 -r requirements.txt
+RUN python -m pip install --extra-index-url https://download.pytorch.org/whl/cu121 torch==2.5.1+cu121
+RUN python -m pip install --no-build-isolation --extra-index-url https://download.pytorch.org/whl/cu121 -r requirements_vorschlag.txt
 
 #RUN python -m pip install flash-attn==2.5.7 --no-build-isolation --extra-index-url https://download.pytorch.org/whl/cu121 -r requirements.txt
 #RUN python -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex/

requirements_vorschlag.txt

@@ -0,0 +1,18 @@
+bitsandbytes
+datasets
+debugpy
+flair==0.14.0
+flash-attn
+huggingface-hub
+hydra-core
+Jinja2
+loguru
+peft
+protobuf
+safetensors
+sentence-transformers==3.3.1
+sentencepiece
+tokenizers
+torch==2.5.1+cu121
+transformers
+uv