holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

StopAsyncIteration should not trigger HTTP500.

Browse Source 
Now returns faster into the ready state.
This commit is contained in:
Marc-Andre Ferland 2022-10-14 05:20:44 -04:00
parent f91c77bdc6
commit 4a7260b1be
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ui/server.py
View File

@ -279,8 +279,10 @@ def save_model_to_config(model_name):
@app.get('/ping') # Get server and optionally session status.
def ping(session_id:str=None):
if current_state_error or not render_thread.is_alive(): # Render thread is dead.
return HTTPException(status_code=500, detail=str(current_state_error))
if not render_thread.is_alive(): # Render thread is dead.
if current_state_error: return HTTPException(status_code=500, detail=str(current_state_error))
return HTTPException(status_code=500, detail='Render thread is dead.')
if current_state_error and not isinstance(current_state_error, StopAsyncIteration): return HTTPException(status_code=500, detail=str(current_state_error))
# Alive
response = {'status': str(current_state)}
if session_id: