Update generate.py

[CodingQinghao]

Description

FIX: Traceback (most recent call last): File "D:\generate.py", line 60, in prompt = BAICHUAN2_PROMPT_FORMAT.format(prompt=args.prompt) NameError: name 'BAICHUAN2_PROMPT_FORMAT' is not defined. Did you mean: 'BAICHUAN_PROMPT_FORMAT'?

1. Why the change?

code error

2. User API changes

NO

3. Summary of the change

• BAICHUAN_PROMPT_FORMAT = "<reserved_106> {prompt} <reserved_107>"

• BAICHUAN2_PROMPT_FORMAT = "<reserved_106> {prompt} <reserved_107>"

4. How to test?

• [ ] N/A
• [ √ ] Unit test
• [ ] Application test
• [ ] Document test
• [ ] ...

5. New dependencies

no dependencies