OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-05-15 16:29:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fallback to pip if uv failes

Browse Source
This commit is contained in:
w-e-w 2025-02-28 23:51:37 +09:00
parent c59a2badd2
commit 9d1accfea0
No known key found for this signature in database
GPG Key ID: 52CFE43196F6280E
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/uv_hook.py
View File

@ -1,4 +1,5 @@
import sys
import copy
import shlex
import subprocess
from functools import wraps
@ -18,10 +19,11 @@ def patch():
@wraps(subprocess.__original_run)
def patched_run(*args, **kwargs):
_kwargs = copy.copy(kwargs)
if args:
command, *_args = args
else:
command, _args = kwargs.pop("args", ""), ()
command, _args = _kwargs.pop("args", ""), ()
if isinstance(command, str):
command = shlex.split(command)
@ -31,7 +33,7 @@ def patch():
assert isinstance(command, list)
if "pip" not in command:
return subprocess.__original_run([*command, *_args], **kwargs)
return subprocess.__original_run([*command, *_args], **_kwargs)
cmd = command[command.index("pip") + 1 :]
@ -40,6 +42,9 @@ def patch():
modified_command = ["uv", "pip", *cmd]
return subprocess.__original_run([*modified_command, *_args], **kwargs)
result = subprocess.__original_run([*modified_command, *_args], **_kwargs)
if result.returncode != 0:
return subprocess.__original_run(*args, **kwargs)
return result
subprocess.run = patched_run