update

common.py

@@ -694,7 +694,7 @@ def text2image_fn(
     pipeline = PIPELINE_IMG if ip_image is None else PIPELINE_IMG_IP
     if ip_image is not None:
         pipeline.set_ip_adapter_scale([ip_adapt_scale])
-    
+
     images = text2img_gen(
         prompt=prompt,
         n_sample=n_sample,

embodied_gen/data/differentiable_render.py

@@ -505,7 +505,7 @@ def parse_args():
         help="Text prompts for the rendering.",
     )
 
-    args = parser.parse_args()
+    args, unknown = parser.parse_known_args()
 
     if args.uuid is None and args.mesh_path is not None:
         args.uuid = []

embodied_gen/scripts/imageto3d.py

@@ -20,7 +20,7 @@ import logging
 import os
 import sys
 from glob import glob
-from shutil import copy, copytree
+from shutil import copy, copytree, rmtree
 
 import numpy as np
 import trimesh
@@ -294,6 +294,8 @@ if __name__ == "__main__":
 
             # Organize the final result files
             result_dir = f"{output_root}/result"
+            if os.path.exists(result_dir):
+                rmtree(result_dir, ignore_errors=True)
             os.makedirs(result_dir, exist_ok=True)
             copy(urdf_path, f"{result_dir}/{os.path.basename(urdf_path)}")
             copytree(