extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-24 19:38:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

Increase task timeout from 15 mins to 30 mins

Browse Source
This commit is contained in:
cmdr2 2023-05-03 18:47:09 +05:30
parent 5c95bcc65d
commit 01c7712961
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ui/easydiffusion/task_manager.py
View File

@ -7,7 +7,7 @@ Notes:
import json import json
import traceback import traceback
TASK_TTL = 15 * 60 # seconds, Discard last session's task timeout TASK_TTL = 30 * 60 # seconds, Discard last session's task timeout
import queue import queue
import threading import threading
@ -398,8 +398,8 @@ def get_devices():
return {"name": device_manager.get_processor_name()} return {"name": device_manager.get_processor_name()}
mem_free, mem_total = torch.cuda.mem_get_info(device) mem_free, mem_total = torch.cuda.mem_get_info(device)
mem_free /= float(10 ** 9) mem_free /= float(10**9)
mem_total /= float(10 ** 9) mem_total /= float(10**9)
return { return {
"name": torch.cuda.get_device_name(device), "name": torch.cuda.get_device_name(device),