Skip to content

Commit 8cccb84

Browse files
feat: rename save_metas_file -> load_metas_file ni join method (#11)
1 parent efb8907 commit 8cccb84

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/update.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -118,13 +118,13 @@ def update_weights(
118118
def join(
119119
ps: ParameterServer,
120120
checkpoint_name: str,
121-
save_metas_file: str,
121+
load_metas_file: str,
122122
req_func: Callable[[list[tuple[str, str]]], None],
123123
inference_parallel_size: int,
124124
endpoint: str,
125125
):
126-
assert save_metas_file, "save_metas_file is required"
127-
with open(save_metas_file, "rb") as f:
126+
assert load_metas_file, "load_metas_file is required"
127+
with open(load_metas_file, "rb") as f:
128128
metas = pickle.load(f)
129129
ps.init_process_group()
130130
check_vllm_ready(endpoint, inference_parallel_size)

0 commit comments

Comments
 (0)