Rename AC-Stream runtime interface
Browse files
README.md
CHANGED
|
@@ -136,7 +136,7 @@ python examples/libero/multigpu_rollout.py \
|
|
| 136 |
--gpus 0,1,2,3 \
|
| 137 |
--suites libero_spatial,libero_object,libero_goal,libero_10 \
|
| 138 |
--num-trials 1 \
|
| 139 |
-
--config examples/libero/configs/recipes/
|
| 140 |
--checkpoint-format fastwam \
|
| 141 |
--checkpoint checkpoints/streamwam/ac-stream/model.pt \
|
| 142 |
--backbone-path "$BACKBONE_PATH" \
|
|
@@ -145,7 +145,7 @@ python examples/libero/multigpu_rollout.py \
|
|
| 145 |
--num-steps-wait 30 \
|
| 146 |
--replan-steps 16 \
|
| 147 |
--num-inference-steps 1 \
|
| 148 |
-
--sampling-method
|
| 149 |
--fixed-seed \
|
| 150 |
--mujoco-gl egl \
|
| 151 |
--save-video
|
|
@@ -153,14 +153,14 @@ python examples/libero/multigpu_rollout.py \
|
|
| 153 |
|
| 154 |
### AC-Stream accelerated
|
| 155 |
|
| 156 |
-
Use the same checkpoint and append `--
|
| 157 |
|
| 158 |
```bash
|
| 159 |
python examples/libero/multigpu_rollout.py \
|
| 160 |
--gpus 0,1,2,3 \
|
| 161 |
--suites libero_spatial,libero_object,libero_goal,libero_10 \
|
| 162 |
--num-trials 1 \
|
| 163 |
-
--config examples/libero/configs/recipes/
|
| 164 |
--checkpoint-format fastwam \
|
| 165 |
--checkpoint checkpoints/streamwam/ac-stream/model.pt \
|
| 166 |
--backbone-path "$BACKBONE_PATH" \
|
|
@@ -169,8 +169,8 @@ python examples/libero/multigpu_rollout.py \
|
|
| 169 |
--num-steps-wait 30 \
|
| 170 |
--replan-steps 16 \
|
| 171 |
--num-inference-steps 1 \
|
| 172 |
-
--sampling-method
|
| 173 |
-
--
|
| 174 |
--fixed-seed \
|
| 175 |
--mujoco-gl egl \
|
| 176 |
--save-video
|
|
|
|
| 136 |
--gpus 0,1,2,3 \
|
| 137 |
--suites libero_spatial,libero_object,libero_goal,libero_10 \
|
| 138 |
--num-trials 1 \
|
| 139 |
+
--config examples/libero/configs/recipes/streamwam_libero_ac_stream_wan22_5b.yaml \
|
| 140 |
--checkpoint-format fastwam \
|
| 141 |
--checkpoint checkpoints/streamwam/ac-stream/model.pt \
|
| 142 |
--backbone-path "$BACKBONE_PATH" \
|
|
|
|
| 145 |
--num-steps-wait 30 \
|
| 146 |
--replan-steps 16 \
|
| 147 |
--num-inference-steps 1 \
|
| 148 |
+
--sampling-method ac-stream \
|
| 149 |
--fixed-seed \
|
| 150 |
--mujoco-gl egl \
|
| 151 |
--save-video
|
|
|
|
| 153 |
|
| 154 |
### AC-Stream accelerated
|
| 155 |
|
| 156 |
+
Use the same checkpoint and append `--ac-stream-accelerated`:
|
| 157 |
|
| 158 |
```bash
|
| 159 |
python examples/libero/multigpu_rollout.py \
|
| 160 |
--gpus 0,1,2,3 \
|
| 161 |
--suites libero_spatial,libero_object,libero_goal,libero_10 \
|
| 162 |
--num-trials 1 \
|
| 163 |
+
--config examples/libero/configs/recipes/streamwam_libero_ac_stream_wan22_5b.yaml \
|
| 164 |
--checkpoint-format fastwam \
|
| 165 |
--checkpoint checkpoints/streamwam/ac-stream/model.pt \
|
| 166 |
--backbone-path "$BACKBONE_PATH" \
|
|
|
|
| 169 |
--num-steps-wait 30 \
|
| 170 |
--replan-steps 16 \
|
| 171 |
--num-inference-steps 1 \
|
| 172 |
+
--sampling-method ac-stream \
|
| 173 |
+
--ac-stream-accelerated \
|
| 174 |
--fixed-seed \
|
| 175 |
--mujoco-gl egl \
|
| 176 |
--save-video
|