Skip to content

Commit d207d0b

Browse files
authored
[wwb] Avoid install auto-gptq/autoawq (openvinotoolkit#3242)
<!-- Keep your pull requests (PRs) as atomic as possible. That increases the likelihood that an individual PR won't be stuck because of adjacent problems, merge conflicts, or code review. Your merged PR is going to appear in the automatically generated release notes on GitHub. So the clearer the title the better. --> ## Description <!-- Please include a summary of the change. Also include relevant motivation and context. --> <!-- Jira ticket number (e.g., 123). Delete if there's no ticket. --> [CVS-178727](https://jira.devtools.intel.com/browse/CVS-178727) <!-- Remove if not applicable --> Fixes #(issue) ## Checklist: - [ ] Tests have been updated or added to cover the new code. <!-- If the change isn't maintenance related, update the tests at https://github.com/openvinotoolkit/openvino.genai/tree/master/tests or explain in the description why the tests don't need an update. --> - [ ] This patch fully addresses the ticket. <!--- If follow-up pull requests are needed, specify in description. --> - [ ] I have made corresponding changes to the documentation. <!-- Run github.com/\<username>/openvino.genai/actions/workflows/deploy_gh_pages.yml on your fork with your branch as a parameter to deploy a test version with the updated content. Replace this comment with the link to the built docs. -->
1 parent 05d994c commit d207d0b

File tree

2 files changed

+4
-15
lines changed

2 files changed

+4
-15
lines changed

tools/who_what_benchmark/requirements.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,6 @@ numpy>=1.23.5
88
tqdm>=4.66.1
99
diffusers
1010
datasets>=3.6.0
11-
auto-gptq; sys_platform == "linux"
12-
autoawq<0.2.8; sys_platform == "linux"
1311
sentencepiece
1412
jinja2>=3.1.0
1513
scipy

tools/who_what_benchmark/tests/test_cli_text.py

Lines changed: 4 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,10 @@
2222
base_model_path = os.path.join(tmp_dir, "opt125m")
2323
target_model_path = os.path.join(tmp_dir, "opt125m_int8")
2424

25-
gptq_model_id = "ybelkada/opt-125m-gptq-4bit"
25+
# awq/gptq models are skipped for now: 180586
2626
awq_model_id = "TitanML/tiny-mixtral-AWQ-4bit"
27+
# model load failed - ticket: 178940, 180586
28+
gptq_model_id = "ybelkada/opt-125m-gptq-4bit"
2729

2830

2931
def setup_module():
@@ -159,20 +161,9 @@ def test_text_language(tmp_path):
159161
assert "马克" in data["prompts"].values[0]
160162

161163

162-
hf_model_scope = [
163-
(model_id),
164-
]
165-
if sys.platform != 'darwin' and sys.platform != 'win32':
166-
hf_model_scope += [
167-
# model load failed in optimum, ticket: 178940
168-
# (gptq_model_id),
169-
(awq_model_id),
170-
]
171-
172-
173164
@pytest.mark.parametrize(
174165
("model_id"),
175-
hf_model_scope,
166+
[(model_id)],
176167
)
177168
def test_text_hf_model(model_id, tmp_path):
178169
temp_file_name = tmp_path / "gt.csv"

0 commit comments

Comments
 (0)