StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

read_info_from_image: add type

Browse Source
This commit is contained in:
Aarni Koskela 2023-07-03 13:10:20 +03:00
parent b2c574891f
commit 96f0593c8f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/images.py
View File

@ -1,3 +1,5 @@
from __future__ import annotations
import datetime import datetime
import pytz import pytz
@ -669,7 +671,7 @@ IGNORED_INFO_KEYS = {
} }
def read_info_from_image(image): def read_info_from_image(image: Image.Image) -> tuple[str | None, dict]:
items = image.info or {} items = image.info or {}
geninfo = items.pop('parameters', None) geninfo = items.pop('parameters', None)