Skip to content

Commit d1d5576

Browse files
authored
[Fix] Update inference-only-demo for InternVLA-N1-DualVLN (#220)
* [Fix] Update inference-only-demo for InternVLA-N1-DualVLN * [Fix] fix notebook output.
1 parent f8331e8 commit d1d5576

File tree

2 files changed

+328
-65
lines changed

2 files changed

+328
-65
lines changed

internnav/agent/internvla_n1_agent_realworld.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -222,10 +222,10 @@ def step_s2(self, rgb, depth, pose, instruction, intrinsic, look_down=False):
222222
**inputs,
223223
max_new_tokens=128,
224224
do_sample=False,
225-
use_cache=True,
226-
past_key_values=self.past_key_values,
225+
# use_cache=True,
226+
# past_key_values=self.past_key_values,
227227
return_dict_in_generate=True,
228-
raw_input_ids=copy.deepcopy(inputs.input_ids),
228+
# raw_input_ids=copy.deepcopy(inputs.input_ids),
229229
)
230230
output_ids = outputs.sequences
231231

@@ -253,5 +253,5 @@ def step_s2(self, rgb, depth, pose, instruction, intrinsic, look_down=False):
253253
return action_seq, None, None
254254

255255
def step_s1(self, latent, rgb, depth):
256-
all_trajs = self.model.generate_traj(latent, rgb, depth, use_async=True)
256+
all_trajs = self.model.generate_traj(latent, rgb, depth)
257257
return all_trajs

0 commit comments

Comments
 (0)