Update README.md

This commit is contained in:
Junyang Lin
2023-08-03 18:24:41 +08:00
committed by GitHub
parent 4610ef397b
commit 278eeb57d1

View File

@@ -79,35 +79,32 @@ Now you can start with ModelScope or Transformers.
To use Qwen-7B-Chat for the inference, all you need to do is to input a few lines of codes as demonstrated below: To use Qwen-7B-Chat for the inference, all you need to do is to input a few lines of codes as demonstrated below:
```python ```python
>>> from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
>>> from transformers.generation import GenerationConfig from transformers.generation import GenerationConfig
>>> tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True)
>>> model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", device_map="auto", trust_remote_code=True).eval() model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", device_map="auto", trust_remote_code=True).eval()
>>> model.generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True) # 可指定不同的生成长度、top_p等相关超参 model.generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True) # 可指定不同的生成长度、top_p等相关超参
>>> # 第一轮对话 1st dialogue turn # 第一轮对话 1st dialogue turn
>>> response, history = model.chat(tokenizer, "你好", history=None) response, history = model.chat(tokenizer, "你好", history=None)
>>> print(response) print(response)
你好很高兴为你提供帮助 # 你好!很高兴为你提供帮助。
>>> # 第二轮对话 2nd dialogue turn
>>> response, history = model.chat(tokenizer, "给我讲一个年轻人奋斗创业最终取得成功的故事。", history=history)
>>> print(response)
这是一个关于一个年轻人奋斗创业最终取得成功的故事
故事的主人公叫李明他来自一个普通的家庭父母都是普通的工人从小李明就立下了一个目标要成为一名成功的企业家 # 第二轮对话 2nd dialogue turn
response, history = model.chat(tokenizer, "给我讲一个年轻人奋斗创业最终取得成功的故事。", history=history)
print(response)
# 这是一个关于一个年轻人奋斗创业最终取得成功的故事。
# 故事的主人公叫李明,他来自一个普通的家庭,父母都是普通的工人。从小,李明就立下了一个目标:要成为一名成功的企业家。
# 为了实现这个目标,李明勤奋学习,考上了大学。在大学期间,他积极参加各种创业比赛,获得了不少奖项。他还利用课余时间去实习,积累了宝贵的经验。
# 毕业后,李明决定开始自己的创业之路。他开始寻找投资机会,但多次都被拒绝了。然而,他并没有放弃。他继续努力,不断改进自己的创业计划,并寻找新的投资机会。
# 最终,李明成功地获得了一笔投资,开始了自己的创业之路。他成立了一家科技公司,专注于开发新型软件。在他的领导下,公司迅速发展起来,成为了一家成功的科技企业。
# 李明的成功并不是偶然的。他勤奋、坚韧、勇于冒险,不断学习和改进自己。他的成功也证明了,只要努力奋斗,任何人都有可能取得成功。
为了实现这个目标李明勤奋学习考上了大学在大学期间他积极参加各种创业比赛获得了不少奖项他还利用课余时间去实习积累了宝贵的经验 # 第三轮对话 3rd dialogue turn
response, history = model.chat(tokenizer, "给这个故事起一个标题", history=history)
毕业后李明决定开始自己的创业之路他开始寻找投资机会但多次都被拒绝了然而他并没有放弃他继续努力不断改进自己的创业计划并寻找新的投资机会 print(response)
# 《奋斗创业:一个年轻人的成功之路》
最终李明成功地获得了一笔投资开始了自己的创业之路他成立了一家科技公司专注于开发新型软件在他的领导下公司迅速发展起来成为了一家成功的科技企业
李明的成功并不是偶然的他勤奋坚韧勇于冒险不断学习和改进自己他的成功也证明了只要努力奋斗任何人都有可能取得成功
>>> # 第三轮对话 3rd dialogue turn
>>> response, history = model.chat(tokenizer, "给这个故事起一个标题", history=history)
>>> print(response)
奋斗创业一个年轻人的成功之路
``` ```