chaojiemao commited on
Commit
a7dec88
·
verified ·
1 Parent(s): dd72440

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -18,6 +18,13 @@ from scepter.modules.transform.io import pillow_convert
18
  from scepter.modules.utils.config import Config
19
  from scepter.modules.utils.distribute import we
20
  from scepter.modules.utils.file_system import FS
 
 
 
 
 
 
 
21
  from examples.examples import fft_examples
22
  from inference.registry import INFERENCES
23
  from inference.utils import edit_preprocess
 
18
  from scepter.modules.utils.config import Config
19
  from scepter.modules.utils.distribute import we
20
  from scepter.modules.utils.file_system import FS
21
+ if os.path.exists('__init__.py'):
22
+ package_name = 'scepter_ext'
23
+ spec = importlib.util.spec_from_file_location(package_name, '__init__.py')
24
+ package = importlib.util.module_from_spec(spec)
25
+ sys.modules[package_name] = package
26
+ spec.loader.exec_module(package)
27
+
28
  from examples.examples import fft_examples
29
  from inference.registry import INFERENCES
30
  from inference.utils import edit_preprocess