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

This commit is contained in:
Michael Goin
2024-06-20 17:00:13 -06:00
committed by GitHub
parent 3f3b6b2150
commit 8065a7e220
22 changed files with 72 additions and 45 deletions
+2 -3
View File
@@ -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',