set sys.path to include "." before loading simplet1i module

This commit is contained in:
Lincoln Stein 2022-08-21 11:03:22 -04:00
commit 7085d1910b

View File

@ -2,6 +2,7 @@ import argparse
import shlex import shlex
import atexit import atexit
import os import os
import sys
# readline unavailable on windows systems # readline unavailable on windows systems
try: try:
@ -34,7 +35,7 @@ def main():
setup_readline() setup_readline()
print("* Initializing, be patient...\n") print("* Initializing, be patient...\n")
os.path.append('.') sys.path.append('.')
from pytorch_lightning import logging from pytorch_lightning import logging
from ldm.simplet2i import T2I from ldm.simplet2i import T2I