Stardust-minus commited on
Commit
9d10593
·
verified ·
1 Parent(s): e3cf3e2
Files changed (1) hide show
  1. tools/webui/inference.py +1 -1
tools/webui/inference.py CHANGED
@@ -54,7 +54,7 @@ def inference_wrapper(
54
  except Exception as e:
55
  er = "CUDA error: device-side assert triggered"
56
  if er in str(e):
57
- app.close()
58
  else:
59
  raise Exception(e)
60
 
 
54
  except Exception as e:
55
  er = "CUDA error: device-side assert triggered"
56
  if er in str(e):
57
+ os._exit(0)
58
  else:
59
  raise Exception(e)
60