From 110c4f70df62c015a3b10b92da0d8cac6336d1e3 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Wed, 21 Sep 2022 11:31:15 -0400 Subject: [PATCH] fix module loading errors 1. Add ldm/dream/restoration/__init__.py file that was inadvertently not committed earlier. 2. Add '.' to sys.path to address weird mac problem reported in #723 --- ldm/dream/restoration/__init__.py | 5 +++++ scripts/dream.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) create mode 100644 ldm/dream/restoration/__init__.py diff --git a/ldm/dream/restoration/__init__.py b/ldm/dream/restoration/__init__.py new file mode 100644 index 0000000000..25b13e8f87 --- /dev/null +++ b/ldm/dream/restoration/__init__.py @@ -0,0 +1,5 @@ +''' +Initialization file for the ldm.dream.restoration package +''' +from .base import Restoration + diff --git a/scripts/dream.py b/scripts/dream.py index dc1757dce2..a11ca17b68 100755 --- a/scripts/dream.py +++ b/scripts/dream.py @@ -8,6 +8,7 @@ import shlex import copy import warnings import time +sys.path.append('.') # corrects a weird problem on Macs import ldm.dream.readline from ldm.dream.args import Args, metadata_dumps, metadata_from_png from ldm.dream.pngwriter import PngWriter @@ -36,7 +37,6 @@ def main(): sys.exit(-1) print('* Initializing, be patient...\n') - sys.path.append('.') from ldm.generate import Generate # these two lines prevent a horrible warning message from appearing