forked from tangger/lerobot
fix for observations with text
This commit is contained in:
@@ -109,7 +109,7 @@ def predict_action(observation, policy, device, use_amp):
|
||||
):
|
||||
# Convert to pytorch format: channel first and float32 in [0,1] with batch dimension
|
||||
for name in observation:
|
||||
if isinstance(observation[name], list):
|
||||
if not isinstance(observation[name], torch.Tensor):
|
||||
continue # VLA-like and multirobot policies include textual inputs in the observation
|
||||
|
||||
if "image" in name:
|
||||
|
||||
Reference in New Issue
Block a user