Fix a bug where the device wouldn't get set if no cuda-compatible hardware was found

This commit is contained in:
cmdr2 2022-10-27 22:51:45 +05:30
parent 01a2fa7c2d
commit 554650c18d
3 changed files with 5 additions and 4 deletions

View File

@ -1,7 +1,7 @@
@echo off
@copy sd-ui-files\scripts\on_env_start.bat scripts\ /Y
@copy sd-ui-files\scripts\bootstrap.bat scripts\ /Y
@REM @copy sd-ui-files\scripts\on_env_start.bat scripts\ /Y
@REM @copy sd-ui-files\scripts\bootstrap.bat scripts\ /Y
if exist "%cd%\profile" (
set USERPROFILE=%cd%\profile

View File

@ -1,7 +1,7 @@
#!/bin/bash
cp sd-ui-files/scripts/on_env_start.sh scripts/
cp sd-ui-files/scripts/bootstrap.sh scripts/
#cp sd-ui-files/scripts/on_env_start.sh scripts/
#cp sd-ui-files/scripts/bootstrap.sh scripts/
# activate the installer env
CONDA_BASEPATH=$(conda info --base)

View File

@ -113,6 +113,7 @@ def device_init(device_selection=None):
return
if not torch.cuda.is_available():
print('WARNING: torch.cuda is not available. Using the CPU, but this will be very slow!')
thread_data.device = 'cpu'
return
device_count = torch.cuda.device_count()
if device_count <= 1 and device_selection == 'auto':