Upload 17 files

model-00001-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ad400726c30db44e20053106096d423ed2c7887b5b1b3be79da8306e27915ec7
-size 135
+oid sha256:10d8636256018712c5e5c823d12e22b5797f99bb721bd123bf6bf2379892be85
+size 6948961960

model-00002-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a32b3491c48a2d9adeadbcbd8820af5f1f34fe18b101934e1959bd663abf5fde
-size 135
+oid sha256:2050b14f21d48904d269f48d5a980fecea87cd7b36641d9b0f015e72d1fe216a
+size 6971232040

model-00003-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fbc11afdbbb2ddbc6cc43c34d76897a6fffbf4c6c539e09ff333964d133c5daa
-size 135
+oid sha256:ea65305a1577f36f721965bf84c8caec0a948ce7ce84d754701637376c531fef
+size 1162406824

modeling_prismatic.py

@@ -517,8 +517,6 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
         # Run VLA inference
         generated_ids = self.generate(input_ids, max_new_tokens=self.get_action_dim(unnorm_key), **kwargs)
 
-        print(generated_ids)
-
         # Extract predicted action tokens and translate into (normalized) continuous actions
         predicted_action_token_ids = generated_ids[0, -self.get_action_dim(unnorm_key) :].cpu().numpy()
         discretized_actions = self.vocab_size - predicted_action_token_ids

tokenizer.model

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1a8f238a200be6c23fbba0f9a999ab4fe3c09ca303b29805e68cf6659bfb7d89
-size 131
+oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
+size 499723