Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes bug when postprocessing image with no metadata

Browse Source
This commit is contained in:
psychedelicious 2022-11-21 14:56:06 +11:00 committed by blessedcoolant
parent a1d80fd106
commit 089c85a017
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
backend/invoke_ai_web_server.py
View File

@ -324,8 +324,9 @@ class InvokeAIWebServer:
)
os.remove(thumbnail_path)
except Exception as e:
traceback.print_exc()
socketio.emit("error", {"message": f"Unable to delete {f}"})
socketio.emit("error", {"message": f"Unable to delete {f}: {str(e)}"})
pass
socketio.emit("tempFolderEmptied")
except Exception as e:
self.socketio.emit("error", {"message": (str(e))})
@ -430,8 +431,9 @@ class InvokeAIWebServer:
"category": category,
}
)
except:
socketio.emit("error", {"message": f"Unable to load {path}"})
except Exception as e:
socketio.emit("error", {"message": f"Unable to load {path}: {str(e)}"})
pass
socketio.emit(
"galleryImages",
@ -500,9 +502,10 @@ class InvokeAIWebServer:
"category": category,
}
)
except:
except Exception as e:
print(f">> Unable to load {path}")
socketio.emit("error", {"message": f"Unable to load {path}"})
socketio.emit("error", {"message": f"Unable to load {path}: {str(e)}"})
pass
socketio.emit(
"galleryImages",
@ -574,8 +577,9 @@ class InvokeAIWebServer:
try:
seed = original_image["metadata"]["image"]["seed"]
except (NameError, AttributeError) as e:
except (KeyError) as e:
seed = "unknown_seed"
pass
if postprocessing_parameters["type"] == "esrgan":
progress.set_current_status("Upscaling (ESRGAN)")