diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py index 231ea887c0..0691cccb7b 100644 --- a/ldm/simplet2i.py +++ b/ldm/simplet2i.py @@ -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)) diff --git a/scripts/dream.py b/scripts/dream.py index f7452d9e48..7ba6205653 100755 --- a/scripts/dream.py +++ b/scripts/dream.py @@ -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')