From bfcac426f5ffba2303128be6b709adad75a68e77 Mon Sep 17 00:00:00 2001
From: vfdev-5 <vfdev.5@gmail.com>
Date: Thu, 13 Nov 2025 10:43:36 +0000
Subject: [PATCH] Updated pyproject.toml

Replaced default torch installation with cpu version only
---
 pyproject.toml | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index ebf5dcd90..dc0a2b63b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -237,3 +237,11 @@ quote-style = "single"
 [tool.uv]
 # Ignore uv.lock and always upgrade the package to the latest
 upgrade-package = ["jax", "jaxlib", "orbax-checkpoint"]
+
+[tool.uv.sources]
+torch = { index = "pytorch" }
+
+[[tool.uv.index]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true