From 13a5288d011495ad8a354ab7d2d0dbd1ccf19e1c Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 10 Jun 2025 21:08:18 -0700
Subject: [PATCH] Fix an error when upgrading from cuda 12.4 to cuda 12.8

---
 one_click.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/one_click.py b/one_click.py
index 94f2aab0..050da76b 100644
--- a/one_click.py
+++ b/one_click.py
@@ -298,7 +298,7 @@ def update_pytorch_and_python():
 
 
 def clean_outdated_pytorch_cuda_dependencies():
-    patterns = ["cu121", "cu122", "torch2.4"]
+    patterns = ["cu121", "cu122", "torch2.4", "torchvision", "torchaudio"]
     result = run_cmd("python -m pip list --format=freeze", capture_output=True, environment=True)
     matching_packages = []