-
Notifications
You must be signed in to change notification settings - Fork 1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[BFCL] Support for pre-existing completion endpoint (#864)
# URL Endpoint Support for BFCL This PR is a product of the discussion in #850. ## Description This PR adds support for using pre-existing OpenAI-compatible endpoints in BFCL, allowing users to bypass the built-in vLLM/sglang server setup. This is particularly useful for distributed environments like SLURM clusters where model serving and benchmarking need to be handled as separate jobs. ## Changes - Added `--skip-server-setup` flag to CLI - Added environment variable support for endpoint configuration: - `VLLM_ENDPOINT` (defaults to 'localhost') - `VLLM_PORT` (defaults to existing VLLM_PORT constant) - Modified OSSHandler to support external endpoints - Updated documentation for new configuration options ## Usage Users can now specify custom endpoints in two ways: 1. Using environment variables: ```bash export VLLM_ENDPOINT="custom.host.com" export VLLM_PORT="8000" ``` 2. Using a `.env` file: ```bash VLLM_ENDPOINT=custom.host.com VLLM_PORT=8000 ``` Then run BFCL with the `--skip-server-setup` flag: ```bash python -m bfcl generate --model MODEL_NAME --backend vllm --skip-server-setup ``` ## Related Issue Closes #850 --------- Co-authored-by: Huanzhi (Hans) Mao <[email protected]>
- Loading branch information
1 parent
5fe4a87
commit 1729c9b
Showing
6 changed files
with
160 additions
and
106 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.