Improve streaming, log_level, memory report, weight loading, and benchmark script (#7632)
Co-authored-by: Kan Wu <wukanustc@gmail.com>
This commit is contained in:
@@ -178,7 +178,7 @@ class TestVisionChunkedPrefill(CustomTestCase):
|
||||
print(output_chunked)
|
||||
print("output without chunked prefill:")
|
||||
print(output_no_chunked)
|
||||
assert output_chunked == output_no_chunked
|
||||
self.assertEqual(output_chunked, output_no_chunked)
|
||||
|
||||
def test_chunked_prefill(self):
|
||||
self._test_chunked_prefill(batches=[False, True], num_frames=[1, [2, 6, 8, 10]])
|
||||
|
||||
Reference in New Issue
Block a user