diff --git a/.gitignore b/.gitignore index 889ab60c84..992cb2be90 100644 --- a/.gitignore +++ b/.gitignore @@ -169,7 +169,7 @@ cython_debug/ # option (not recommended) you can uncomment the following to ignore the entire idea folder. #.idea/ -src/ +src logs/ **/__pycache__/ outputs diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000000..c8f0748436 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,13 @@ +[submodule "taming-transformers"] + path = src/taming-transformers + url = https://github.com/CompVis/taming-transformers.git + ignore = dirty +[submodule "clip"] + path = src/clip + url = https://github.com/openai/CLIP.git + ignore = dirty +[submodule "k-diffusion"] + path = src/k-diffusion + url = https://github.com/lstein/k-diffusion.git + ignore = dirty + \ No newline at end of file diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py index 09141c8ebb..be6c05bdc9 100644 --- a/ldm/simplet2i.py +++ b/ldm/simplet2i.py @@ -68,6 +68,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 @@ -416,7 +417,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/requirements.txt b/requirements.txt index 79f5fee300..78d43aae26 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,5 @@ accelerate==0.12.0 albumentations==0.4.3 -clip==1.0 einops==0.3.0 huggingface-hub==0.8.1 imageio==2.9.0 diff --git a/scripts/dream.py b/scripts/dream.py index 886d8bd682..dca26b16a0 100755 --- a/scripts/dream.py +++ b/scripts/dream.py @@ -81,8 +81,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')