Add get weights by parameter name for llama (#2266)

2024-11-29 23:36:38 -08:00
parent 7d5d1d3d29
commit 7d1485d376
12 changed files with 337 additions and 17 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -38,6 +38,7 @@ suites = {
        "test_update_weights.py",
        "test_vision_openai_server.py",
        "test_session_control.py",
+        "test_get_parameter_by_name.py",
    ],
    "sampling/penaltylib": glob.glob(
        "sampling/penaltylib/**/test_*.py", recursive=True