fix: pin torch 2.1.0 and compatible versions to avoid torchao conflict

Dockerfile

@@ -3,20 +3,19 @@ FROM python:3.10-slim
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV MPLCONFIGDIR=/tmp/matplotlib
-ENV PIP_CACHE_DIR=/tmp/pip-cache
 ENV HOME=/tmp
 
 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 --index-url https://download.pytorch.org/whl/cu121
+RUN pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu121
 
-RUN pip install --no-cache-dir trl peft accelerate bitsandbytes wandb datasets huggingface_hub transformers
+RUN pip install --no-cache-dir transformers==4.36.0 accelerate==0.25.0 peft==0.7.1 bitsandbytes==0.41.3 trl==0.7.4
 
 RUN pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps
 
-RUN pip install --no-cache-dir unsloth_zoo
+RUN pip install --no-cache-dir unsloth_zoo wandb datasets huggingface_hub
 
 RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git