From 273eb9f9974136bd8a72bbc7d7969904e4cce2bd Mon Sep 17 00:00:00 2001 From: nayan Date: Sat, 25 Jul 2026 18:30:29 -0400 Subject: [PATCH] whatever --- openpilot/sunnypilot/modeld_v2/compile_modeld.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/openpilot/sunnypilot/modeld_v2/compile_modeld.py b/openpilot/sunnypilot/modeld_v2/compile_modeld.py index 200ad1db1b..d4ea49bb45 100755 --- a/openpilot/sunnypilot/modeld_v2/compile_modeld.py +++ b/openpilot/sunnypilot/modeld_v2/compile_modeld.py @@ -148,14 +148,13 @@ def create_jit_runner(vision_runner, policy_runners: list, nv12: NV12Frame, mode inputs['traffic_convention'] = traffic_conv_dev if vision_runner: - vision_out = next(iter(vision_runner({road_key: img, wide_key: big_img}).values())) - vision_out_cast = vision_out.cast('float32') + vision_out_cast = next(iter(vision_runner({road_key: img, wide_key: big_img}).values())).cast('float32').realize() new_feat = vision_out_cast[:, features_slice].reshape(1, -1).unsqueeze(0) inputs['features_buffer'] = shift_and_sample(feat_q, new_feat, sample_skip_fn).realize() - policy_outs = [next(iter(pol_runner(inputs).values())).cast('float32') for pol_runner in policy_runners] + policy_outs = [next(iter(pol_runner(inputs).values())).cast('float32').realize() for pol_runner in policy_runners] return (vision_out_cast, *policy_outs) if len(policy_outs) > 1 else (vision_out_cast, policy_outs[0]) inputs.update({road_key: img, wide_key: big_img, 'features_buffer': sample_skip_fn(feat_q)}) - policy_out = next(iter(policy_runners[0](inputs).values())).cast('float32') + policy_out = next(iter(policy_runners[0](inputs).values())).cast('float32').realize() new_feat = policy_out[:, features_slice].reshape(1, -1).unsqueeze(0) shift_and_sample(feat_q, new_feat, sample_skip_fn).realize() return policy_out