update dep

app.py

@@ -35,10 +35,13 @@ 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@master',
+        f'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()
 
@@ -53,10 +56,6 @@ from guided_diffusion.script_util import (
     control_net_defaults,
 )
 
-th.backends.cuda.matmul.allow_tf32 = True
-th.backends.cudnn.allow_tf32 = True
-th.backends.cudnn.enabled = True
-
 from pathlib import Path
 
 from tqdm import tqdm, trange
@@ -202,7 +201,7 @@ def main(args):
                                     eval_data=None,
                                     **vars(args))
     
-    @spaces.GPU()
+    @spaces.GPU(duration=200)
     def reconstruct_and_export(*args, **kwargs):
         return training_loop_class.eval_i23d_and_export(*args, **kwargs)