MiniWan commited on
Commit
b3e4c42
·
verified ·
1 Parent(s): 3a0762d

Update model card

Browse files
Files changed (1) hide show
  1. README.md +1 -3
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
  ```