Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
app.py
CHANGED
@@ -207,7 +207,7 @@ def inference(req: ServeTTSRequest):
|
|
207 |
|
208 |
except Exception as e:
|
209 |
er = "CUDA error: device-side assert triggered"
|
210 |
-
if er in e:
|
211 |
app.close()
|
212 |
else:
|
213 |
raise Exception(e)
|
|
|
207 |
|
208 |
except Exception as e:
|
209 |
er = "CUDA error: device-side assert triggered"
|
210 |
+
if er in str(e):
|
211 |
app.close()
|
212 |
else:
|
213 |
raise Exception(e)
|