From 7ccd94ed4fe12092e3c4ff659e3bea649f6f8a4c Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Tue, 10 Jun 2025 13:59:32 +0300
Subject: [PATCH] Force upgrade pip when installing Torch

---
 CHANGELOG.md | 1 +
 installer.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 1593fbf57..d6b7f0cd6 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -14,6 +14,7 @@
 - **Torch**  
   - Set default to `torch==2.7.1`  
   - Support ROCm 6.4 with `---use-nightly`  
+  - Force upgrade pip when installing Torch  
 
 - **SDNQ Quantization**  
   - Add group size support for convolutional layers  
diff --git a/installer.py b/installer.py
index cf45e3737..0769808e9 100644
--- a/installer.py
+++ b/installer.py
@@ -869,6 +869,7 @@ def check_torch():
     if 'torch' in torch_command and not args.version:
         if not installed('torch'):
             log.info(f'Torch: download and install in progress... cmd="{torch_command}"')
+            install('--upgrade pip', 'pip', reinstall=True) # pytorch rocm is too large for older pip
         install(torch_command, 'torch torchvision', quiet=True)
     else:
         try: