set char_pos to zero

app.py

@@ -121,6 +121,7 @@ def generate(
     padding_mask = out["padding_mask"][0].to(bool).cpu()
     padded_traj = out["gen_samples"][0].cpu()
     traj = padded_traj[padding_mask]
+    char_traj = out["char_feat"][0].cpu()
     padded_vertices = out["char_raw"]["char_vertices"][0]
     vertices = padded_vertices[padding_mask]
     faces = out["char_raw"]["char_faces"][0]
@@ -134,6 +135,7 @@ def generate(
     log_sample(
         root_name="world",
         traj=traj.numpy(),
+        char_traj=char_traj.numpy(),
         K=K,
         vertices=vertices.numpy(),
         faces=faces.numpy(),

utils/common_viz.py

@@ -101,6 +101,7 @@ def get_batch(
     # Update batch
     batch["caption_raw"] = [prompt]
     batch["caption_feat"] = caption_feat
+    batch['char_feat'] = torch.zeros_like(batch['char_feat'])
 
     return batch