|
|
@@ -247,14 +247,8 @@ fn collect_nccl_env<'a>(ctx: &'a DockerCommandCtx, node: &'a Node) -> Vec<(&'a s
|
|
|
|
|
|
/// Build the list of `-e KEY=VALUE` Docker env var pairs for a node.
|
|
|
fn collect_docker_env(ctx: &DockerCommandCtx, node: &Node, is_head: bool) -> Vec<String> {
|
|
|
- let mut pairs: Vec<(&str, Cow<str>)> = vec![
|
|
|
- ("HF_HOME", Cow::Borrowed("/root/.cache/huggingface")),
|
|
|
- // Wrapper-style entrypoints (e.g. sparkring's serve-with-warmup) poll
|
|
|
- // the API at 127.0.0.1:$PORT; mirror the recipe port so recipes stay
|
|
|
- // free of a redundant `PORT` env entry. Inert for stock vllm/sglang
|
|
|
- // (they take `--port` as a CLI flag, which the launcher still appends).
|
|
|
- ("PORT", Cow::Owned(ctx.recipe.port.to_string())),
|
|
|
- ];
|
|
|
+ let mut pairs: Vec<(&str, Cow<str>)> =
|
|
|
+ vec![("HF_HOME", Cow::Borrowed("/root/.cache/huggingface"))];
|
|
|
if ctx.nnodes > 1 {
|
|
|
pairs.extend(collect_nccl_env(ctx, node));
|
|
|
}
|