[core] minimum torch version is 2.6 (#13725)

minimum torch version is 2.6

Author: sayakpaul

docker/diffusers-pytorch-minimum-cuda/Dockerfile

@@ -4,9 +4,9 @@ LABEL repository="diffusers"
 
 ARG PYTHON_VERSION=3.10
 ENV DEBIAN_FRONTEND=noninteractive
-ENV MINIMUM_SUPPORTED_TORCH_VERSION="2.1.0"
-ENV MINIMUM_SUPPORTED_TORCHVISION_VERSION="0.16.0"
-ENV MINIMUM_SUPPORTED_TORCHAUDIO_VERSION="2.1.0"
+ENV MINIMUM_SUPPORTED_TORCH_VERSION="2.6.0"
+ENV MINIMUM_SUPPORTED_TORCHVISION_VERSION="0.21.0"
+ENV MINIMUM_SUPPORTED_TORCHAUDIO_VERSION="2.6.0"
 
 RUN apt-get -y update \
     && apt-get install -y software-properties-common \

docs/source/en/installation.md

@@ -12,7 +12,7 @@ specific language governing permissions and limitations under the License.
 
 # Installation
 
-Diffusers is tested on Python 3.8+ and PyTorch 1.4+. Install [PyTorch](https://pytorch.org/get-started/locally/) according to your system and setup.
+Diffusers is tested on Python 3.8+ and PyTorch 2.6+. Install [PyTorch](https://pytorch.org/get-started/locally/) according to your system and setup.
 
 Create a [virtual environment](https://packaging.python.org/guides/installing-using-pip-and-virtual-environments/) for easier management of separate projects and to avoid compatibility issues between dependencies. Use [uv](https://docs.astral.sh/uv/), a Rust-based Python package and project manager, to create a virtual environment and install Diffusers.
 

setup.py

@@ -137,7 +137,7 @@
     "requests",
     "tensorboard",
     "tiktoken>=0.7.0",
-    "torch>=1.4",
+    "torch>=2.6",
     "torchvision",
     "transformers>=4.41.2",
     "urllib3<=2.0.0",

src/diffusers/dependency_versions_table.py

@@ -44,7 +44,7 @@
     "requests": "requests",
     "tensorboard": "tensorboard",
     "tiktoken": "tiktoken>=0.7.0",
-    "torch": "torch>=1.4",
+    "torch": "torch>=2.6",
     "torchvision": "torchvision",
     "transformers": "transformers>=4.41.2",
     "urllib3": "urllib3<=2.0.0",