Skip to content

Commit 6e80d19

Browse files
authored
Test optimized generation longbench trust_remote_code=true (#2492)
Fails on MAC: https://github.com/openvinotoolkit/openvino.genai/actions/runs/16562135998/job/46853033964#step:7:221
1 parent 454e4d1 commit 6e80d19

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/python_tests/test_kv_cache_eviction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ def test_optimized_generation_longbench(test_struct):
220220
generation_config.num_return_sequences = 1
221221
generation_config.max_new_tokens = max_new_tokens
222222

223-
data = datasets.load_dataset('THUDM/LongBench', subset, split='test[:32]')
223+
data = datasets.load_dataset('THUDM/LongBench', subset, split='test[:32]', trust_remote_code=True)
224224
with tqdm(total=len(data)) as progress_bar:
225225
batch = []
226226
answers = []

0 commit comments

Comments
 (0)