Fix torchao int1 error by pinning to 0.6.1

Dockerfile

@@ -11,8 +11,9 @@ WORKDIR /app
 
 # Upgrade pip and install unsloth properly
 RUN pip install --no-cache-dir --upgrade pip
-RUN pip install --no-cache-dir "torchao<=0.7.0"
+RUN pip install --no-cache-dir "torchao==0.6.1"
 RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
+RUN pip install --no-cache-dir trl
 
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt

requirements.txt

@@ -1,6 +1,5 @@
 fastapi
 uvicorn[standard]
-unsloth
 pytest
 ruff
 mypy