10-models_deepseek_v4_1_nvidia_model_state.py.diff 471 B

1234567891011
  1. --- baked-ours/models/deepseek_v4_1/nvidia/model_state.py 2026-09-10 20:13:20.000000000 +0200
  2. +++ baked/models/deepseek_v4_1/nvidia/model_state.py 2026-09-11 10:13:47.000000000 +0200
  3. @@ -113,6 +113,8 @@
  4. input_batch.query_start_loc[: input_batch.num_reqs + 1],
  5. window,
  6. input_batch.num_tokens,
  7. + input_batch=input_batch,
  8. + req_states=req_states,
  9. )
  10. return model_inputs