Skip to content

Commit

Permalink
Add Python backend when vLLM backend built (#6397)
Browse files Browse the repository at this point in the history
  • Loading branch information
dyastremsky authored and krishung5 committed Oct 10, 2023
1 parent f04d9de commit bbc1715
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions build.py
Original file line number Diff line number Diff line change
Expand Up @@ -2504,6 +2504,15 @@ def enable_all():
log('backend "{}" at tag/branch "{}"'.format(parts[0], parts[1]))
backends[parts[0]] = parts[1]

if "vllm" in backends:
if "python" not in backends:
log(
"vLLM backend requires Python backend, adding Python backend with tag {}".format(
backends["vllm"]
)
)
backends["python"] = backends["vllm"]

# Initialize map of repo agents to build and repo-tag for each.
repoagents = {}
for be in FLAGS.repoagent:
Expand Down

0 comments on commit bbc1715

Please sign in to comment.