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