Fix import, add pip install

#2
by pcuenq HF Staff - opened
Files changed (1) hide show
  1. README.md +5 -1
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.pi0.modeling_pi0 import PI0FastPolicy
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)