File size: 3,735 Bytes
fab7c2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
# MiniCPM
## 介绍 Introduction
- 与`Llama`的关系 The Relationship between `Llama`
`MiniCPM`与`Llama`均使用了仅解码器架构。代码实现上,`MiniCPM`基于`Llama`实现,增加了放缩机制。
`MiniCPM` uses Decoder-only Structure as well as `Llama`. The implementation of `MiniCPM` is based on `Llama` code, with scaling mechenism added.
## 软件依赖 Dependency
- `transformers >= 4.36.0`
- `accelerate`
## 使用 Usage
我们推荐使用`AutoModelForCausalLM`与`AutoTokenizer`载入`MiniCPM`,并使用`torch.bfloat16`作为计算精度。我们推荐在GPU上进行推理。
We recommend using `AutoModelForCausalLM` and `AutoTokenizer` to load `MiniCPM`, and use `torch.bfloat16` as the calculation precision. GPU reference is recommended.
以下是一个使用`MiniCPM`生成的例子。
An example is provided below for using `MiniCPM` to generate tokens.
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
path = '/data/miniCPM_opensource/miniCPM-bf16' # TODO
tokenizer = AutoTokenizer.from_pretrained(path)
model = AutoModelForCausalLM.from_pretrained(path, torch_dtype=torch.bfloat16, device_map='auto', trust_remote_code=True)
dialog = [{'role': 'user', 'content': '请问中国哪几个城市最适合旅游?'}]
input = tokenizer.apply_chat_template(dialog, tokenize=False, add_generation_prompt=False)
enc = tokenizer(input, return_tensors='pt').to('cuda')
output = model.generate(**enc, max_length=1024)
print(tokenizer.decode(output[0]))
```
期望的输出 Expected Output:
```
<s> <用户>请问中国哪几个城市最适合旅游?<AI> 中国有很多适合旅游的城市,以下是一些建议:
1. 北京:作为中国的首都,北京拥有丰富的历史文化遗产,如故宫、长城、天坛等。此外,北京还有许多现代化的景点,如798艺术区、颐和园等。
2. 上海:作为中国的经济中心,上海拥有许多现代化的高楼大厦和繁华的商业区。同时,上海还有许多历史悠久的景点,如外滩、豫园等。
3. 西安:作为古都,西安拥有丰富的历史文化遗产,如兵马俑、大雁塔等。此外,西安还有许多美食,如肉夹馍、羊肉泡馍等。
4. 成都:作为四川的省会,成都有着丰富的美食文化,如火锅、麻辣烫等。此外,成都还有许多自然风光,如青城山、都江堰等。
5. 杭州:作为美丽的西湖所在地,杭州拥有许多自然风光和历史文化遗产,如西湖、灵隐寺等。此外,杭州还有许多美食,如西湖醋鱼、龙井虾仁等。
6. 广州:作为南方的重要城市,广州拥有丰富的美食文化,如广式早茶、烧腊等。此外,广州还有许多现代化景点,如珠江夜游、白云山等。
7. 南京:作为六朝古都,南京拥有丰富的历史文化遗产,如中山陵、夫子庙等。此外,南京还有许多美食,如鸭血粉丝汤、盐水鸭等。
8. 厦门:作为美丽的海滨城市,厦门拥有许多自然风光和历史文化遗产,如鼓浪屿、南普陀寺等。此外,厦门还有许多美食,如沙茶面、土笋冻等。
9. 昆明:作为云南的省会,昆明拥有许多自然风光和历史文化遗产,如石林、滇池等。此外,昆明还有许多美食,如过桥米线、酸笋鱼等。
10. 哈尔滨:作为北方的城市,哈尔滨拥有许多冰雪景观和自然风光,如冰雪大世界、亚布力滑雪场等。此外,哈尔滨还有许多美食,如东北大拉皮、锅包肉等。
以上仅是一些建议,中国还有许多其他适合旅游的城市,具体取决于您的兴趣和偏好。</s>
```
## 引用 Reference |