Spaces:

abreza
/

et-viewer

Sleeping

azizinaghsh commited on Dec 13, 2024

Commit

2868f4f

1 Parent(s): bbcccf9

fix typo

Files changed (1) hide show

utils/common_viz.py CHANGED Viewed

@@ -103,7 +103,7 @@ def get_batch(
     batch["caption_raw"] = [prompt]
     batch["caption_feat"] = caption_feat
-    character_Position = torch.tensor(
         [float(coord) for coord in character_position.strip("[]").split(",")],
         device=device,
     )

     batch["caption_raw"] = [prompt]
     batch["caption_feat"] = caption_feat
+    character_position = torch.tensor(
         [float(coord) for coord in character_position.strip("[]").split(",")],
         device=device,
     )