Add update_weights_from_tensor (#2631)

2024-12-29 05:30:27 +08:00
parent 7863e4368a
commit fd28640dc5
10 changed files with 120 additions and 1 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -40,6 +40,7 @@ suites = {
        "test_triton_attention_kernels.py",
        "test_triton_attention_backend.py",
        "test_update_weights_from_disk.py",
+        "test_update_weights_from_tensor.py",
        "test_vision_chunked_prefill.py",
        "test_vision_openai_server.py",
        "test_session_control.py",