holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

device_selection is already a string, since we've used string functions before this line

Browse Source
This commit is contained in:
cmdr2 2022-11-07 18:08:43 +05:30
parent abbfae2fc0
commit 90b1609d4e
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ui/sd_internal/runtime.py
View File

@ -129,7 +129,9 @@ def device_init(device_selection=None):
thread_data.force_full_precision = False thread_data.force_full_precision = False
thread_data.reduced_memory = True thread_data.reduced_memory = True
if device_selection.lower() == 'cpu': device_selection = device_selection.lower()
if device_selection == 'cpu':
thread_data.device = 'cpu' thread_data.device = 'cpu'
thread_data.device_name = get_processor_name() thread_data.device_name = get_processor_name()
print('Render device CPU available as', thread_data.device_name) print('Render device CPU available as', thread_data.device_name)
@ -162,10 +164,10 @@ def device_init(device_selection=None):
print(f'Setting GPU:{device} as active') print(f'Setting GPU:{device} as active')
torch.cuda.device(device) torch.cuda.device(device)
return return
if isinstance(device_selection, str):
device_selection = device_selection.lower() if device_selection.startswith('gpu:'):
if device_selection.startswith('gpu:'): device_selection = int(device_selection[4:])
device_selection = int(device_selection[4:])
if device_selection != 'cuda' and device_selection != 'current' and device_selection != 'gpu': if device_selection != 'cuda' and device_selection != 'current' and device_selection != 'gpu':
if device_select(device_selection): if device_select(device_selection):
if isinstance(device_selection, int): if isinstance(device_selection, int):