community[patch]: fix yuan2 errors in LLMs (#19004)

1. fix yuan2 errors while invoke Yuan2. 2. update tests.
2025-09-24 12:01:54 +00:00 · 2024-03-29 05:37:44 +08:00
parent aba4bd0d13
commit b7c8bc8268
2 changed files with 16 additions and 5 deletions
--- a/libs/community/tests/integration_tests/llms/test_yuan2.py
+++ b/libs/community/tests/integration_tests/llms/test_yuan2.py
@@ -11,7 +11,6 @@ def test_yuan2_call_method() -> None:
        max_tokens=1024,
        temp=1.0,
        top_p=0.9,
-        top_k=40,
        use_history=False,
    )
    output = llm("写一段快速排序算法。")
@@ -25,7 +24,6 @@ def test_yuan2_generate_method() -> None:
        max_tokens=1024,
        temp=1.0,
        top_p=0.9,
-        top_k=40,
        use_history=False,
    )
    output = llm.generate(["who are you?"])