python代码片段
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--temp', type=float, default=0.9)
parser.add_argument('--topk', type=int, default=50)
parser.add_argument('--lm', type=str, default='7')
args = parser.parse_args()
temp=args.temp
llama_type=args.lm
top_k=args.topk
终端
python generate.py --temp 0.9 --lm 13