update dep

app.py

@@ -32,56 +32,6 @@ import numpy as np
 import torch as th
 import torch.distributed as dist
 
-def install_dependency():
-    # install full cuda first
-    subprocess.run(
-        f'conda install -c nvidia cuda-nvcc',
-        shell=True
-    )
-
-    # install apex
-    subprocess.run(
-        f'TORCH_CUDA_ARCH_LIST="compute capability" FORCE_CUDA=1 {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@master',
-        shell=True,
-    )
-
-th.backends.cuda.matmul.allow_tf32 = True
-th.backends.cudnn.allow_tf32 = True
-th.backends.cudnn.enabled = True
-
-install_dependency()
-
-from guided_diffusion import dist_util, logger
-from guided_diffusion.script_util import (
-    NUM_CLASSES,
-    model_and_diffusion_defaults,
-    create_model_and_diffusion,
-    add_dict_to_argparser,
-    args_to_dict,
-    continuous_diffusion_defaults,
-    control_net_defaults,
-)
-
-from pathlib import Path
-
-from tqdm import tqdm, trange
-import dnnlib
-from nsr.train_util_diffusion import TrainLoop3DDiffusion as TrainLoop
-from guided_diffusion.continuous_diffusion import make_diffusion as make_sde_diffusion
-import nsr
-import nsr.lsgm
-from nsr.script_util import create_3DAE_model, encoder_and_nsr_defaults, loss_defaults, AE_with_Diffusion, rendering_options_defaults, eg3d_options_default, dataset_defaults
-
-from datasets.shapenet import load_eval_data
-from torch.utils.data import Subset
-from datasets.eg3d_dataset import init_dataset_kwargs
-
-from transport.train_utils import parse_transport_args
-
-from utils.infer_utils import remove_background, resize_foreground
-
-SEED = 0
-
 
 def resize_to_224(img):
     img = transforms.functional.resize(img, 224,
@@ -477,6 +427,61 @@ if __name__ == "__main__":
 
     @spaces.GPU
     def greet(n):
+
+        def install_dependency():
+            # install full cuda first
+            # subprocess.run(
+            #     f'conda install -c nvidia cuda-nvcc',
+            #     shell=True
+            # )
+
+            # install apex
+            subprocess.run(
+                f'TORCH_CUDA_ARCH_LIST="compute capability" FORCE_CUDA=1 {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@master',
+                shell=True,
+            )
+
+        th.backends.cuda.matmul.allow_tf32 = True
+        th.backends.cudnn.allow_tf32 = True
+        th.backends.cudnn.enabled = True
+
+        install_dependency()
+
+        from guided_diffusion import dist_util, logger
+        from guided_diffusion.script_util import (
+            NUM_CLASSES,
+            model_and_diffusion_defaults,
+            create_model_and_diffusion,
+            add_dict_to_argparser,
+            args_to_dict,
+            continuous_diffusion_defaults,
+            control_net_defaults,
+        )
+
+        from pathlib import Path
+
+        from tqdm import tqdm, trange
+        import dnnlib
+        from nsr.train_util_diffusion import TrainLoop3DDiffusion as TrainLoop
+        from guided_diffusion.continuous_diffusion import make_diffusion as make_sde_diffusion
+        import nsr
+        import nsr.lsgm
+        from nsr.script_util import create_3DAE_model, encoder_and_nsr_defaults, loss_defaults, AE_with_Diffusion, rendering_options_defaults, eg3d_options_default, dataset_defaults
+
+        from datasets.shapenet import load_eval_data
+        from torch.utils.data import Subset
+        from datasets.eg3d_dataset import init_dataset_kwargs
+
+        from transport.train_utils import parse_transport_args
+
+        from utils.infer_utils import remove_background, resize_foreground
+
+        SEED = 0
+
+
+
+
+
         print(zero.device) # <-- 'cuda:0' 🤗
         return f"Hello {zero + n} Tensor"