udate weights from disk (#2265)

2024-11-29 17:17:00 -08:00
parent b53d6cbda3
commit 7d5d1d3d29
11 changed files with 54 additions and 40 deletions
--- a/test/srt/test_data_parallelism.py
+++ b/test/srt/test_data_parallelism.py
@@ -44,7 +44,7 @@ class TestDataParallelism(unittest.TestCase):

    def test_update_weight(self):
        response = requests.post(
-            self.base_url + "/update_weights",
+            self.base_url + "/update_weights_from_disk",
            json={"model_path": DEFAULT_MODEL_NAME_FOR_TEST},
        )

@@ -55,7 +55,7 @@ class TestDataParallelism(unittest.TestCase):
        time.sleep(5)

        response = requests.post(
-            self.base_url + "/update_weights",
+            self.base_url + "/update_weights_from_disk",
            json={"model_path": DEFAULT_MODEL_NAME_FOR_TEST},
        )