update

app.py

@@ -369,6 +369,8 @@ def main(args):
             """
         )
 
+        gr.Markdown(_CITE_)
+
         submit.click(fn=check_input_image, inputs=[input_image]).success(
             fn=preprocess,
             inputs=[input_image, preprocess_background],

configs/i23d_args.json

@@ -11,7 +11,7 @@
   "ddpm_model_path": "",
   "cldm_model_path": "",
   "rec_model_path": "",
-  "logdir": "./logs/LSGM/inference/Objaverse/i23d/dit-L2/",
+  "logdir": "./",
   "data_dir": "NONE",
   "eval_data_dir": "/cpfs01/user/lanyushi.p/Repo/eccv24/open-source/InstantMesh/test_dir",
   "eval_batch_size": 1,

nsr/train_util_diffusion.py

@@ -331,7 +331,8 @@ class TrainLoopDiffusionWithRec(TrainLoop):
                     ],
                     dim=-1)  # B, 3, H, W
 
-            if save_img:
+            if False:
+            # if save_img:
                 for batch_idx in range(gen_img.shape[0]):
                     sampled_img = Image.fromarray(
                         (gen_img[batch_idx].permute(1, 2, 0).cpu().numpy() *