From 295fb5e7ea3a409dbc4e84ca40289e8940ed5a08 Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Thu, 22 Jun 2023 23:50:32 -0400
Subject: [PATCH] chore(cli): normalize kebab case (#50)

---
 src/openllm/cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/openllm/cli.py b/src/openllm/cli.py
index 2603d7b4..d70061df 100644
--- a/src/openllm/cli.py
+++ b/src/openllm/cli.py
@@ -772,7 +772,7 @@ start = functools.partial(_start, _serve_grpc=False)
 start_grpc = functools.partial(_start, _serve_grpc=True)
 
 
-@cli.command()
+@cli.command(context_settings={"token_normalize_func": inflection.underscore})
 @click.argument(
     "model_name", type=click.Choice([inflection.dasherize(name) for name in openllm.CONFIG_MAPPING.keys()])
 )