Skip to content

Commit 50cd102

Browse files
committed
fix
Signed-off-by: wang.yuqi <[email protected]>
1 parent 00d377b commit 50cd102

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

examples/offline_inference/converting2seq_cls_models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,8 @@ def converting(model_name, classifier_from_tokens, path, method, device="cpu"):
7777

7878
def parse_args():
7979
parser = argparse.ArgumentParser(
80-
description="Converting *ForCausalLM models to *ForSequenceClassification models."
80+
description="Converting *ForCausalLM models to "
81+
"*ForSequenceClassification models."
8182
)
8283
parser.add_argument(
8384
"--model_name", type=str, default="Qwen/Qwen3-Reranker-0.6B", help="Model name"

tests/models/language/pooling/score_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,5 +52,6 @@ def ping_pong_test_score_models(hf_runner,
5252
]
5353

5454
for i in range(len(sentences)):
55-
assert float(hf_outputs[i]) == pytest.approx(float(vllm_outputs[i]), rel=0.01), \
55+
assert float(hf_outputs[i]) == pytest.approx(float(vllm_outputs[i]),
56+
rel=0.01), \
5657
f"Test failed at #{i}, vllm: {vllm_outputs[i]}, st: {hf_outputs[i]}"

0 commit comments

Comments
 (0)