-
Notifications
You must be signed in to change notification settings - Fork 205
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add LongBench validation #1220
base: master
Are you sure you want to change the base?
Add LongBench validation #1220
Conversation
6c96bd0
to
eefe4f2
Compare
is it still valid? |
000d2a4
to
7eb1e54
Compare
6c3e950
to
becdf78
Compare
d71206d
to
15d3c77
Compare
LongBenchTestData("trec", 3.2, 2.0, 3.3), | ||
LongBenchTestData("qasper", 5.8, 1.7, 3.6), | ||
]) | ||
def test_optimized_generation_longbench(qwen2_converted_model, test_struct): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
is it explicitly not marked as @pytest.mark.precommit
?
how long does this test take?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes
On my local machine it takes 24 minutes, but on CI it takes about 60 minutes.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we run this test in nightly jobs?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
yes, please add mark for nightly
@@ -0,0 +1,245 @@ | |||
import re |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This code is probably taken from 3rd party repo. We need to keep the license in this case.
15d3c77
to
1e88d10
Compare
# Copyright (C) 2023-2025 Intel Corporation | ||
# SPDX-License-Identifier: Apache-2.0 | ||
# | ||
# This file includes utility functions copied from the LongBench repository: | ||
# https://github.com/THUDM/LongBench | ||
# | ||
# Copyright (c) 2023 THU-KEG & Zhipu AI | ||
# Licensed under the MIT License |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@moslex please take a look at the license. Could you confirm if it is sufficient for code adapted from a 3rd party repo?
No description provided.