[Frontend] Add FlexibleArgumentParser to support both underscore and dash in names (#5718)

2024-06-20 19:00:13 -04:00
parent 3f3b6b2150
commit 8065a7e220
22 changed files with 72 additions and 45 deletions
--- a/examples/aqlm_example.py
+++ b/examples/aqlm_example.py
@@ -1,11 +1,10 @@
-import argparse
-
 from vllm import LLM, SamplingParams
+from vllm.utils import FlexibleArgumentParser


 def main():

-    parser = argparse.ArgumentParser(description='AQLM examples')
+    parser = FlexibleArgumentParser(description='AQLM examples')

    parser.add_argument('--model',
                        '-m',