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
This commit is contained in:
Lincoln Stein 2022-09-21 11:31:15 -04:00
parent 19174949b6
commit 110c4f70df
2 changed files with 6 additions and 1 deletions

View File

@ -0,0 +1,5 @@
'''
Initialization file for the ldm.dream.restoration package
'''
from .base import Restoration

View File

@ -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