Update README.md
Browse files
README.md
CHANGED
@@ -51,8 +51,9 @@ model = create_model(
|
|
51 |
control_frequency=control_frequency,
|
52 |
)
|
53 |
# Start inference process
|
|
|
54 |
lang_embeddings_path = 'your/language/embedding/path'
|
55 |
-
text_embedding = torch.load(lang_embeddings_path)['embeddings']
|
56 |
images: List(PIL.Image) = ... # The images from last 2 frame
|
57 |
proprio = ... # The current robot state
|
58 |
# Perform inference to predict the next chunk_size actions
|
|
|
51 |
control_frequency=control_frequency,
|
52 |
)
|
53 |
# Start inference process
|
54 |
+
# Load pre-computed language embeddings
|
55 |
lang_embeddings_path = 'your/language/embedding/path'
|
56 |
+
text_embedding = torch.load(lang_embeddings_path)['embeddings']
|
57 |
images: List(PIL.Image) = ... # The images from last 2 frame
|
58 |
proprio = ... # The current robot state
|
59 |
# Perform inference to predict the next chunk_size actions
|