Merge pull request #885 from TinyBeeman/outputStats

Output stats
This commit is contained in:
ZeroCool 2022-09-09 01:51:51 -07:00 committed by GitHub
commit 3781f97b0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1862,7 +1862,7 @@ def layout():
save_grid, group_by_prompt, save_as_jpg, use_GFPGAN, use_RealESRGAN, RealESRGAN_model, fp=defaults.general.fp,
variant_amount=variant_amount, variant_seed=variant_seed, write_info_files=write_info_files)
message.success('Done!', icon="")
message.success('Render Complete: ' + info + '; Stats: ' + stats, icon="")
except KeyError:
output_images, seed, info, stats = txt2img(prompt, st.session_state.sampling_steps, sampler_name, RealESRGAN_model, batch_count, 1,
@ -1870,7 +1870,7 @@ def layout():
save_grid, group_by_prompt, save_as_jpg, use_GFPGAN, use_RealESRGAN, RealESRGAN_model, fp=defaults.general.fp,
variant_amount=variant_amount, variant_seed=variant_seed, write_info_files=write_info_files)
message.success('Done!', icon="")
message.success('Render Complete: ' + info + '; Stats: ' + stats, icon="")
except (StopException):
print(f"Received Streamlit StopException")
@ -2018,7 +2018,7 @@ def layout():
)
#show a message when the generation is complete.
message.success('Done!', icon="")
message.success('Render Complete: ' + info + '; Stats: ' + stats, icon="")
except (StopException, KeyError):
print(f"Received Streamlit StopException")