extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-27 10:43:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Process GFPGANer on cuda:0 when possible, otherwise use cpu.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-16 23:12:46 -04:00
parent 1d4c5cc96f
commit d4a348a2b2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/sd_internal/task_manager.py
View File

@ -217,7 +217,8 @@ def thread_render(device):
if not queued_task.request.use_cpu and runtime.thread_data.device == 'cpu': if not queued_task.request.use_cpu and runtime.thread_data.device == 'cpu':
continue # CPU Tasks continue # CPU Tasks
if queued_task.request.use_face_correction and not runtime.is_first_cuda_device(runtime.thread_data.device): if queued_task.request.use_face_correction and not runtime.is_first_cuda_device(runtime.thread_data.device):
continue #TODO Remove when fixed - A bug with GFPGANer and facexlib needs to be fixed before use on other devices. if not runtime.thread_data.device == 'cpu' and is_alive(0) > 0: # Allows GFPGANer on cuda:0 and use cpu only when cuda:0 is not available.
continue #TODO Remove when fixed - A bug with GFPGANer and facexlib needs to be fixed before use on other devices.
task = queued_task task = queued_task
break break
if task is not None: if task is not None: