Update Dockerfile

Dockerfile

@@ -1,17 +1,14 @@
-FROM nvidia/cuda:12.9.0-devel-ubuntu24.04
+FROM nvidia/cuda:12.9.0-runtime-ubuntu24.04
 
 # Avoid interactive prompts during package installation
 ENV DEBIAN_FRONTEND=noninteractive
 
 # Install Python and system dependencies
-# devel image needed for flash_attn compilation (Granite NAR requires it)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3 \
     python3-pip \
     python3-dev \
     git \
-    gcc \
-    g++ \
     libsndfile1 \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
@@ -34,18 +31,15 @@ RUN pip install --no-cache-dir \
 
 # Install common requirements (torch already installed above, pip will skip it)
 RUN pip install --no-cache-dir \
-    transformers==4.57.6 \
+    transformers==4.38 \
     evaluate \
     datasets \
     librosa \
     jiwer \
     num2words \
-    peft==0.13.1 \
+    peft \
     soundfile
 
-# Install Flash Attention 2 (required by Granite NAR model)
-RUN MAX_JOBS=8 pip install --no-cache-dir flash-attn==2.8.3 --no-build-isolation
-
 # Force soundfile backend for datasets audio decoding
 ENV HF_AUDIO_DECODER_BACKEND=soundfile