add torch type override

This commit is contained in:
Vladimir Mandic
2023-05-11 20:36:34 -04:00
parent 5905fcb97e
commit df2cfd5100
2 changed files with 17 additions and 5 deletions
+2 -1
View File
@@ -28,12 +28,13 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
with queue_lock:
shared.state.begin()
progress.start_task(id_task)
res = None
res = [None, '', '', '']
try:
res = func(*args, **kwargs)
progress.record_results(id_task, res)
except Exception as e:
shared.log.error(f"Exception: {e}")
res[-1] = f"<div class='error'>{html.escape(str(e))}</div>"
finally:
progress.finish_task(id_task)
shared.state.end()