doevent commited on
Commit
ea0bf2f
·
verified ·
1 Parent(s): 45eadd5

Fix CUDA error

Browse files

CUDA error: device-side assert triggered CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -546,4 +546,4 @@ if __name__ == "__main__":
546
  logger.info("Warming up done, launching the web UI...")
547
 
548
  app = build_app()
549
- app.launch(show_api=True)
 
546
  logger.info("Warming up done, launching the web UI...")
547
 
548
  app = build_app()
549
+ app.queue(api_open=True).launch(show_error=True, show_api=True)