Update model card
Browse files
README.md
CHANGED
|
@@ -29,14 +29,12 @@ Install the [code](https://github.com/Prism-WAM/PrismWAM), then from the reposit
|
|
| 29 |
|
| 30 |
```bash
|
| 31 |
hf download THUSI-Lab/PrismWAM --include "robotwin/*" --local-dir weights
|
| 32 |
-
hf download Wan-AI/Wan2.2-TI2V-5B --include "models_t5_umt5-xxl-enc-bf16.pth" "google/umt5-xxl/*" \
|
| 33 |
-
--local-dir weights/Wan-AI/Wan2.2-TI2V-5B # text encoder
|
| 34 |
```
|
| 35 |
|
| 36 |
```python
|
| 37 |
from prismwam import load_pretrained
|
| 38 |
|
| 39 |
-
policy = load_pretrained("robotwin")
|
| 40 |
observation["prompt"] = policy.format_prompt("beat the block with the hammer")
|
| 41 |
out = policy.predict(observation, seed=0) # out["action"]: 32 x 14
|
| 42 |
```
|
|
|
|
| 29 |
|
| 30 |
```bash
|
| 31 |
hf download THUSI-Lab/PrismWAM --include "robotwin/*" --local-dir weights
|
|
|
|
|
|
|
| 32 |
```
|
| 33 |
|
| 34 |
```python
|
| 35 |
from prismwam import load_pretrained
|
| 36 |
|
| 37 |
+
policy = load_pretrained("robotwin") # also fetches the T5 text encoder on first use
|
| 38 |
observation["prompt"] = policy.format_prompt("beat the block with the hammer")
|
| 39 |
out = policy.predict(observation, seed=0) # out["action"]: 32 x 14
|
| 40 |
```
|