Fix import, add pip install
#2
by
pcuenq
HF Staff
- opened
README.md
CHANGED
|
@@ -13,11 +13,15 @@ It was proposed in [FAST: Efficient Action Tokenization for Vision-Language-Acti
|
|
| 13 |
|
| 14 |
## How to Get Started
|
| 15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
```python
|
| 17 |
import torch
|
| 18 |
from lerobot.policies.factory import make_pre_post_processors
|
| 19 |
import numpy as np
|
| 20 |
-
from lerobot.policies.
|
| 21 |
|
| 22 |
model_id = "lerobot/pi0fast-base"
|
| 23 |
model = PI0FastPolicy.from_pretrained(model_id)
|
|
|
|
| 13 |
|
| 14 |
## How to Get Started
|
| 15 |
|
| 16 |
+
```bash
|
| 17 |
+
pip install "lerobot[pi]@git+https://github.com/huggingface/lerobot.git"
|
| 18 |
+
```
|
| 19 |
+
|
| 20 |
```python
|
| 21 |
import torch
|
| 22 |
from lerobot.policies.factory import make_pre_post_processors
|
| 23 |
import numpy as np
|
| 24 |
+
from lerobot.policies.pi0_fast.modeling_pi0_fast import PI0FastPolicy
|
| 25 |
|
| 26 |
model_id = "lerobot/pi0fast-base"
|
| 27 |
model = PI0FastPolicy.from_pretrained(model_id)
|