From e215bd676398c18f369179d7c7db10cad582bc8f Mon Sep 17 00:00:00 2001
From: Art Gourieff <85128026+Gourieff@users.noreply.github.com>
Date: Wed, 6 Mar 2024 01:27:24 +0700
Subject: [PATCH] HOTFIX: Rollback to ORT --extra-index-url if CU12

+VersionUP (0.7.0 beta5)
---
 README.md                  | 2 +-
 README_RU.md               | 2 +-
 install.py                 | 3 ++-
 scripts/reactor_version.py | 2 +-
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 7e207f7..1ec67aa 100644
--- a/README.md
+++ b/README.md
@@ -2,7 +2,7 @@
 
   <img src="https://github.com/Gourieff/Assets/raw/main/sd-webui-reactor/ReActor_logo_NEW_EN.png?raw=true" alt="logo" width="180px"/>
     
-  ![Version](https://img.shields.io/badge/version-0.7.0_beta4-green?style=for-the-badge&labelColor=darkgreen)
+  ![Version](https://img.shields.io/badge/version-0.7.0_beta5-green?style=for-the-badge&labelColor=darkgreen)
   
   <a href="https://boosty.to/artgourieff" target="_blank">
     <img src="https://lovemet.ru/www/boosty.jpg" width="108" alt="Support Me on Boosty"/>
diff --git a/README_RU.md b/README_RU.md
index 9370343..6e398de 100644
--- a/README_RU.md
+++ b/README_RU.md
@@ -2,7 +2,7 @@
 
   <img src="https://github.com/Gourieff/Assets/raw/main/sd-webui-reactor/ReActor_logo_NEW_RU.png?raw=true" alt="logo" width="180px"/>
     
-  ![Version](https://img.shields.io/badge/версия-0.7.0_beta4-green?style=for-the-badge&labelColor=darkgreen)
+  ![Version](https://img.shields.io/badge/версия-0.7.0_beta5-green?style=for-the-badge&labelColor=darkgreen)
   
   <a href="https://boosty.to/artgourieff" target="_blank">
     <img src="https://lovemet.ru/www/boosty.jpg" width="108" alt="Поддержать проект на Boosty"/>
diff --git a/install.py b/install.py
index e9f7c38..815a8c3 100644
--- a/install.py
+++ b/install.py
@@ -120,7 +120,8 @@ with open(req_file) as file:
         if cuda_version is not None and float(cuda_version)>=12: # CU12
             if not is_installed(ort,"1.17.1",False):
                 install_count += 1
-                pip_install(ort,"-U")
+                pip_uninstall("onnxruntime", "onnxruntime-gpu")
+                pip_install(ort,"--extra-index-url","https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/")
         elif not is_installed(ort,"1.16.1",False):
             install_count += 1
             pip_install(ort, "-U")
diff --git a/scripts/reactor_version.py b/scripts/reactor_version.py
index 12aa118..695a098 100644
--- a/scripts/reactor_version.py
+++ b/scripts/reactor_version.py
@@ -1,5 +1,5 @@
 app_title = "ReActor"
-version_flag = "v0.7.0-b4"
+version_flag = "v0.7.0-b5"
 
 from scripts.reactor_logger import logger, get_Run, set_Run
 from scripts.reactor_globals import DEVICE