fix: simplify dockerfile no version pinning

Dockerfile

@@ -3,36 +3,20 @@ FROM python:3.10-slim
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV HOME=/tmp
+ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_cache
 
 RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-RUN pip install --no-cache-dir \
-    torch==2.4.0 \
-    --index-url https://download.pytorch.org/whl/cu124
-
-RUN pip install --no-cache-dir \
-    transformers==4.45.0 \
-    accelerate==0.34.2 \
-    peft==0.12.0 \
-    trl==0.12.0 \
-    bitsandbytes==0.44.0 \
-    wandb \
-    datasets \
-    huggingface_hub \
-    matplotlib \
-    Pillow
-
-RUN pip install --no-cache-dir \
-    git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
-
-RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/ && \
-    mkdir -p /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/ && \
-    mkdir -p /app/data/
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu124
 
+RUN pip install --no-cache-dir transformers accelerate peft trl bitsandbytes wandb datasets huggingface_hub matplotlib Pillow
+
+RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
+
+RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/
 COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
-COPY data/ /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/
 COPY data/ /app/data/
 COPY train.py .
 COPY evaluate.py .