Unverified Commit 34af2007 authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #98 from deepghs/dev/meta 

dev(narugo): add empty metadata support for a41 webui images
parents a1720fd1 c02b2b23

imgutils/sd/metadata.py

+7 −1
Original line number Diff line number Diff line
@@ -195,7 +195,13 @@ def parse_sdmeta_from_text(x: str) -> SDMetaData: 
        <class 'imgutils.sd.metadata.SDMetaData'>

    """

    x = textwrap.dedent(x).strip()

    *prompt_lines, argument_line = x.splitlines(keepends=False)

    all_lines = x.splitlines(keepends=False)

    if len(all_lines) >= 1:

        prompt_lines = all_lines[:-1]

        argument_line = all_lines[-1]

    else:

        prompt_lines = []

        argument_line = ''

    if len(_PARAM_PATTERN.findall(argument_line)) < 3:

        prompt_lines.append(argument_line)

        argument_line = ''

test/sd/test_metadata.py

+3 −0
Original line number Diff line number Diff line
@@ -397,3 +397,6 @@ Steps: 20, Sampler: DPM++ 2M SDE Karras, CFG scale: 7, Seed: 2647703743, Size: 7 
            assert get_sdmeta_from_image(f) == sdimg_3_std

        with _test_pnginfo(sdimg_4_std.pnginfo) as f:

            assert get_sdmeta_from_image(f) == sdimg_4_std



    def test_empty_info_parse(self):

        assert parse_sdmeta_from_text('') == SDMetaData('', '', {})