Skip to content

Commit 20272f2

Browse files
committed
fix(vllm): sanitize model reference in backend error logging
1 parent 6ca1a5a commit 20272f2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pkg/inference/scheduling/runner.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import (
1414
"time"
1515

1616
"github.com/docker/model-runner/pkg/inference"
17+
"github.com/docker/model-runner/pkg/internal/utils"
1718
"github.com/docker/model-runner/pkg/logging"
1819
"github.com/docker/model-runner/pkg/metrics"
1920
)
@@ -182,7 +183,7 @@ func run(
182183
go func() {
183184
if err := backend.Run(runCtx, socket, modelID, modelRef, mode, runnerConfig); err != nil {
184185
log.Warnf("Backend %s running model %s exited with error: %v",
185-
backend.Name(), modelRef, err,
186+
backend.Name(), utils.SanitizeForLog(modelRef), err,
186187
)
187188
r.err = err
188189
}

0 commit comments

Comments
 (0)