From 87cde07e21cf4d697b5bda987a5ef4d8691275c8 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 11 Feb 2025 16:56:02 +0100
Subject: [PATCH] Rolling back torch version.

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index d79bb30f..63d11845 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -48,7 +48,7 @@ FROM nvidia/cuda:12.4.1-devel-ubuntu22.04 AS pytorch-install
 WORKDIR /usr/src/
 
 # NOTE: When updating PyTorch version, beware to remove `pip install nvidia-nccl-cu12==2.22.3` below in the Dockerfile. Context: https://github.com/huggingface/text-generation-inference/pull/2099
-ARG PYTORCH_VERSION=2.6.0
+ARG PYTORCH_VERSION=2.5.1
 ARG PYTHON_VERSION=3.11
 
 # Keep in sync with `server/pyproject.toml