new

app.py

@@ -223,7 +223,7 @@ def bundle_image_to_mesh(
     ):
     global mesh_cache
     print(f"Before bundle_image_to_mesh: {torch.cuda.memory_allocated() / 1024**3} GB")
-
+    k3d_wrapper.recon_model.init_flexicubes_geometry("cuda:0", fovy=50.0)
     # TODO: delete this later
     k3d_wrapper.del_llm_model()
 

pipeline/kiss3d_wrapper.py

@@ -142,7 +142,6 @@ def init_wrapper_from_config(config_path):
     state_dict = {k[14:]: v for k, v in state_dict.items() if k.startswith('lrm_generator.')}
     recon_model.load_state_dict(state_dict, strict=True)
     recon_model.to(recon_device)
-    recon_model.init_flexicubes_geometry(recon_device, fovy=50.0)
     recon_model.eval()
     logger.warning(f"GPU memory allocated after load reconstruction model on {recon_device}: {torch.cuda.memory_allocated(device=recon_device) / 1024**3} GB")