[AXK1] Fix expected logits for CUDA A10G (#47980)

[AXK1] Fix expected logits for CUDA A10G in test_model_logits_batched

The expected values for ("cuda", (8, 6)) were incorrect since the model
was added. Measured and updated on A10G (CUDA compute capability 8.6).

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>