We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent efb8907 commit 8cccb84Copy full SHA for 8cccb84
examples/update.py
@@ -118,13 +118,13 @@ def update_weights(
118
def join(
119
ps: ParameterServer,
120
checkpoint_name: str,
121
- save_metas_file: str,
+ load_metas_file: str,
122
req_func: Callable[[list[tuple[str, str]]], None],
123
inference_parallel_size: int,
124
endpoint: str,
125
):
126
- assert save_metas_file, "save_metas_file is required"
127
- with open(save_metas_file, "rb") as f:
+ assert load_metas_file, "load_metas_file is required"
+ with open(load_metas_file, "rb") as f:
128
metas = pickle.load(f)
129
ps.init_process_group()
130
check_vllm_ready(endpoint, inference_parallel_size)
0 commit comments