Spaces:
Runtime error
Runtime error
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 |
|