61 Commits

Author	SHA1	Message	Date
cmdr2	cb82170187	Fix the bug where custom models weren't getting picked up	2022-10-17 21:22:05 +05:30
Marc-Andre Ferland	c92129ac63	Improved detection of missing cuda:0 and added warning to console about how to fix.	2022-10-17 03:32:23 -04:00
Marc-Andre Ferland	554b67a2f0	Fixing bug in is_alive.	2022-10-17 01:05:51 -04:00
Marc-Andre Ferland	012243a880	Process GPU tasks on CPU when there are no cuda devices at all.	2022-10-17 01:05:27 -04:00
Marc-Andre Ferland	d4a348a2b2	Process GFPGANer on cuda:0 when possible, otherwise use cpu.	2022-10-16 23:12:46 -04:00
Marc-Andre Ferland	41bfb96b6b	Fixed bug in task_manager.is_alive and added way to check for first device.	2022-10-16 23:06:41 -04:00
Marc-Andre Ferland	7c72608e1c	First draft for Multi-GPU support	2022-10-16 21:41:39 -04:00
Marc-Andre Ferland	cbdf03450d	Added timeout to critical locking tasks with matching exception	2022-10-15 05:31:17 -04:00
Marc-Andre Ferland	8fdb1e7ec9	Improved locking and logging when cleaning old cached sessions.	2022-10-15 04:39:45 -04:00
Marc-Andre Ferland	d3b28c42e6	Better error handling with cache.put	2022-10-15 04:08:17 -04:00
Marc-Andre Ferland	7de699c7fa	Moved a lot of code into task_manager.py	2022-10-15 03:28:20 -04:00