Update README.md
Browse files
README.md
CHANGED
|
@@ -76,7 +76,8 @@ model = create_model(
|
|
| 76 |
|
| 77 |
# Start inference process
|
| 78 |
# Load the pre-computed language embeddings
|
| 79 |
-
|
|
|
|
| 80 |
text_embedding = torch.load(lang_embeddings_path)['embeddings']
|
| 81 |
images: List(PIL.Image) = ... # The images from last 2 frames
|
| 82 |
proprio = ... # The current robot state
|
|
|
|
| 76 |
|
| 77 |
# Start inference process
|
| 78 |
# Load the pre-computed language embeddings
|
| 79 |
+
# Refer to scripts/encode_lang.py for how to encode the language instruction
|
| 80 |
+
lang_embeddings_path = 'your/language/embedding/path'
|
| 81 |
text_embedding = torch.load(lang_embeddings_path)['embeddings']
|
| 82 |
images: List(PIL.Image) = ... # The images from last 2 frames
|
| 83 |
proprio = ... # The current robot state
|