extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-01 23:56:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use torchruntime with Forge

Browse Source
This commit is contained in:
cmdr2 2025-03-11 11:03:48 +05:30
parent 5285e8f5e8
commit 4044d696e6
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ui/easydiffusion/backends/webui/__init__.py
View File

@ -93,12 +93,9 @@ def install_backend():
# clone webui
run_in_conda(["git", "clone", WEBUI_REPO, WEBUI_DIR], cwd=ROOT_DIR, env=env)
# install cpu-only torch if the PC doesn't have a graphics card (for Windows and Linux).
# this avoids WebUI installing a CUDA version and trying to activate it
torch_platform_name = get_installed_torch_platform()[0]
if OS_NAME in ("Windows", "Linux") and is_cpu_device(torch_platform_name):
run_in_conda(["python", "-m", "pip", "install", "torch", "torchvision"], cwd=WEBUI_DIR, env=env)
# install the appropriate version of torch using torchruntime
run_in_conda(["python", "-m", "pip", "install", "torchruntime"], cwd=WEBUI_DIR, env=env)
run_in_conda(["python", "-m", "torchruntime", "install", "torch", "torchvision"], cwd=WEBUI_DIR, env=env)
def start_backend():