From 835cd3847cca590491dca4447e07afc1444e1fc9 Mon Sep 17 00:00:00 2001
From: Maximilien Le Clei <34255811+MaximilienLC@users.noreply.github.com>
Date: Sun, 23 Feb 2025 14:57:42 -0500
Subject: [PATCH] test rm

---
 pyproject.toml | 2 --
 1 file changed, 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index a01c7194..4da68e81 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,7 +51,6 @@ dependencies = [
     # OPTIONAL for cneuromax/fitting/deeplearning/
     "torchaudio==2.6.0+cu124", # Tensor manipulation on audio data
     "torchvision==0.21.0+cu124", # Tensor manipulation on vision data
-    "torchao==0.8.0+cu124", # Lower precision training/inference
     "transformers==4.49.0", # Pre-trained models published on Hugging Face
     "diffusers==0.32.2", # Diffusion models published on Hugging Face
     "timm==1.0.15", # Image models
@@ -97,7 +96,6 @@ dependencies = [
 torch = { index = "pytorch" }
 torchaudio = { index = "pytorch" }
 torchvision = { index = "pytorch" }
-torchao = { index = "pytorch" }
 
 [[tool.uv.index]]
 name = "pytorch"