update

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/models/text_model.py

@@ -94,6 +94,7 @@ def build_text2img_ip_pipeline(
     pipe.set_ip_adapter_scale([ref_scale])
 
     pipe = pipe.to(device)
+    pipe.image_encoder = pipe.image_encoder.to(device)
     # pipe.enable_model_cpu_offload()
     # pipe.enable_xformers_memory_efficient_attention()
     # pipe.enable_vae_slicing()

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(