Unverified Commit 488f831d authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #5012 from Nandaka/master

Support NAI style exif in PNG Info for Send... buttons
parents 9ec0a41a 904121fe
Loading
Loading
Loading
Loading
+14 −0
Original line number Diff line number Diff line
@@ -234,6 +234,20 @@ def run_pnginfo(image):

    geninfo = items.get('parameters', geninfo)

    # nai prompt
    if "Software" in items.keys() and items["Software"] == "NovelAI":
        import json
        json_info = json.loads(items["Comment"])
        geninfo = f'{items["Description"]}\r\nNegative prompt: {json_info["uc"]}\r\n'
        sampler = "Euler a"
        if json_info["sampler"] == "k_euler_ancestral":
            sampler = "Euler a"
        elif json_info["sampler"] == "k_euler":
            sampler = "Euler"
        model_hash = '925997e9'  # assuming this is the correct model hash
        # not sure with noise and strength parameter
        geninfo += f'Steps: {json_info["steps"]}, Sampler: {sampler}, CFG scale: {json_info["scale"]}, Seed: {json_info["seed"]}, Size: {image.width}x{image.height}, Model hash: {model_hash}'  # , Denoising strength: {json_info["noise"]}'

    info = ''
    for key, text in items.items():
        info += f"""