robotics-diffusion-transformer commited on
Commit
7e4f2e7
·
verified ·
1 Parent(s): 5aa1106

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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'] # Pre-computed language 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