update

app.py

@@ -32,6 +32,16 @@ import numpy as np
 import torch as th
 import torch.distributed as dist
 
+def install_dependency():
+    # install apex
+    subprocess.run(
+        f'{sys.executable} -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git@https://github.com/NVIDIA/apex.git',
+        shell=True,
+    )
+
+
+install_dependency()
+
 from guided_diffusion import dist_util, logger
 from guided_diffusion.script_util import (
     NUM_CLASSES,
@@ -67,13 +77,6 @@ from utils.infer_utils import remove_background, resize_foreground
 
 SEED = 0
 
-def install_dependency():
-    # install apex
-    subprocess.run(
-        f'{sys.executable} -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git@https://github.com/NVIDIA/apex.git',
-        shell=True,
-    )
-
 
 def resize_to_224(img):
     img = transforms.functional.resize(img, 224,
@@ -460,7 +463,6 @@ if __name__ == "__main__":
 
     args.rendering_kwargs = rendering_options_defaults(args)
 
-    install_dependency()
 
     # main(args)