update

app.py

@@ -109,7 +109,7 @@ def main(args):
 
     # args.rendering_kwargs = rendering_options_defaults(args)
 
-    dist_util.setup_dist(args)
+    # dist_util.setup_dist(args)
     logger.configure(dir=args.logdir)
 
     th.cuda.empty_cache()
@@ -466,7 +466,6 @@ if __name__ == "__main__":
     # args = create_argparser().parse_args()
 
     # args.local_rank = int(os.environ["LOCAL_RANK"])
-    # args.local_rank = 0
     # args.gpus = th.cuda.device_count()
 
     # args.rendering_kwargs = rendering_options_defaults(args)
@@ -474,6 +473,9 @@ if __name__ == "__main__":
         args = json.load(f)
         args = EasyDict(args)
 
+    args.local_rank = 0
+    args.gpus = 1
+
 
     # ! demo 
     # zero = torch.Tensor([0]).cuda()