[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>
