mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-01-23 06:52:42 -05:00
fix: raises error if backend is not supported
Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
@@ -1048,6 +1048,8 @@ def build_command(
|
||||
serialisation, default='safetensors' if has_safetensors_weights(model_id, model_version) else 'legacy'
|
||||
),
|
||||
)
|
||||
if llm.__llm_backend__ not in llm.config['backend']:
|
||||
raise click.ClickException(f"'{backend}' is not supported with {model_id}")
|
||||
backend_warning(llm.__llm_backend__, build=True)
|
||||
os.environ.update(
|
||||
**process_environ(
|
||||
|
||||
Reference in New Issue
Block a user