mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
runtime errors now produce a stack trace
This commit is contained in:
parent
f55f4d7156
commit
47a5da25b7
@ -69,6 +69,7 @@ from contextlib import contextmanager, nullcontext
|
|||||||
import time
|
import time
|
||||||
import math
|
import math
|
||||||
import re
|
import re
|
||||||
|
import traceback
|
||||||
|
|
||||||
from ldm.util import instantiate_from_config
|
from ldm.util import instantiate_from_config
|
||||||
from ldm.models.diffusion.ddim import DDIMSampler
|
from ldm.models.diffusion.ddim import DDIMSampler
|
||||||
@ -415,7 +416,8 @@ The vast majority of these arguments default to reasonable values.
|
|||||||
print('*interrupted*')
|
print('*interrupted*')
|
||||||
print('Partial results will be returned; if --grid was requested, nothing will be returned.')
|
print('Partial results will be returned; if --grid was requested, nothing will be returned.')
|
||||||
except RuntimeError as e:
|
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()
|
toc = time.time()
|
||||||
print(f'{image_count} images generated in',"%4.2fs"% (toc-tic))
|
print(f'{image_count} images generated in',"%4.2fs"% (toc-tic))
|
||||||
|
@ -80,7 +80,6 @@ def main():
|
|||||||
exit(-1)
|
exit(-1)
|
||||||
|
|
||||||
# preload the model
|
# 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)...")
|
print("\n* Initialization done! Awaiting your command (-h for help, 'q' to quit, 'cd' to change output dir, 'pwd' to print output dir)...")
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user