We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 695adc7 commit 0b301b7Copy full SHA for 0b301b7
extension/llm/export/test/test_builder.py
@@ -88,7 +88,7 @@ def test_get_dynamic_shape_with_dynamic_shape_enabled_with_kv_cache(self) -> Non
88
# Check first element (tokens dimension)
89
self.assertIsInstance(result[0], dict)
90
self.assertIn(1, result[0])
91
- self.assertEqual(result[0][1].max, self.max_seq_len - 1)
+ self.assertEqual(result[0][1].max, self.max_seq_len)
92
93
# Check second element (input_pos dimension)
94
self.assertIsInstance(result[1], dict)
0 commit comments