adding uv

webcoderz · Mar 19, 2024 · e40aed6 · e40aed6
1 parent d7999a5
commit e40aed6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/modules/launch_utils.py b/modules/launch_utils.py
@@ -30,6 +30,7 @@
     os.environ['GRADIO_ANALYTICS_ENABLED'] = 'False'
 
 
+
 def check_python_version():
     is_windows = platform.system() == "Windows"
     major = sys.version_info.major
@@ -135,8 +136,11 @@ def run_pip(command, desc=None, live=default_command_live):
         return
 
     index_url_line = f' --index-url {index_url}' if index_url != '' else ''
-    return run(f'"{python}" -m pip {command} --prefer-binary{index_url_line}', desc=f"Installing {desc}", errdesc=f"Couldn't install {desc}", live=live)
 
+    if "ENABLE_UV" not in os.environ:
+        return run(f'"{python}" -m pip {command} --prefer-binary{index_url_line}', desc=f"Installing {desc}", errdesc=f"Couldn't install {desc}", live=live)
+    else:
+        return run(f'uv pip {command} --prefer-binary{index_url_line} --no-cache-dir', desc=f"Installing {desc}", errdesc=f"Couldn't install {desc}", live=live)
 
 def check_run_python(code: str) -> bool:
     result = subprocess.run([python, "-c", code], capture_output=True, shell=False)

diff --git a/webui-user.sh b/webui-user.sh
@@ -57,4 +57,5 @@
 #export RAY_ADDRESS='ray://ray-head:10001'
 #export RAY_DOCKER="True"
 #export RAY_DASHBOARD_PORT=8265
-#export RAY_DASHBOARD_HOST="0.0.0.0"
+#export RAY_DASHBOARD_HOST="0.0.0.0"
+export ENABLE_UV="True"