delinqu commited on
Commit
b988a0f
·
verified ·
1 Parent(s): 4f714fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -52,7 +52,7 @@ prompt = "What action should the robot take to pick the cup?"
52
  inputs = processor(images=[image], text=prompt, return_tensors="pt")
53
  generation_outputs = model.predict_action(inputs)
54
 
55
- actions = processor.decode_actions(generation_outputs, unnorm_key="bridge_orig/1.0.0")
56
  print(actions)
57
  ```
58
 
 
52
  inputs = processor(images=[image], text=prompt, return_tensors="pt")
53
  generation_outputs = model.predict_action(inputs)
54
 
55
+ actions = processor.decode_actions(generation_outputs, unnorm_key="fractal20220817_data/0.1.0")
56
  print(actions)
57
  ```
58