Spaces:
Running
on
L4
Running
on
L4
fix exit
Browse files- 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 |
-
|
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 |
|