[hotfix] fix typo s/get_defualt_parser /get_default_parser (#5548)

This commit is contained in:
digger yu
2024-04-07 19:04:58 +08:00
committed by GitHub
parent a799ca343b
commit 341263df48
3 changed files with 5 additions and 5 deletions

View File

@@ -33,7 +33,7 @@ def inference(model, tokenizer, text, **generate_kwargs):
return outputs[0].tolist()
def get_defualt_parser():
def get_default_parser():
parser = argparse.ArgumentParser()
parser.add_argument("--pretrained", type=str, default="hpcaitech/grok-1")
parser.add_argument("--tokenizer", type=str, default="tokenizer.model")