runtime errors now produce a stack trace

This commit is contained in:
Lincoln Stein 2022-08-24 12:57:04 -04:00
parent f55f4d7156
commit 47a5da25b7
2 changed files with 4 additions and 3 deletions

View File

@ -69,6 +69,7 @@ from contextlib import contextmanager, nullcontext
import time
import math
import re
import traceback
from ldm.util import instantiate_from_config
from ldm.models.diffusion.ddim import DDIMSampler
@ -415,7 +416,8 @@ The vast majority of these arguments default to reasonable values.
print('*interrupted*')
print('Partial results will be returned; if --grid was requested, nothing will be returned.')
except RuntimeError as e:
print(str(e))
print("Oops! A runtime error has occurred. If this is unexpected, please copy-and-paste this stack trace and post it as an Issue to http://github.com/lstein/stable-diffusion")
traceback.print_exc()
toc = time.time()
print(f'{image_count} images generated in',"%4.2fs"% (toc-tic))

View File

@ -80,8 +80,7 @@ def main():
exit(-1)
# preload the model
if not debugging:
t2i.load_model()
t2i.load_model()
print("\n* Initialization done! Awaiting your command (-h for help, 'q' to quit, 'cd' to change output dir, 'pwd' to print output dir)...")
log_path = os.path.join(opt.outdir,'dream_log.txt')