svjack commited on
Commit
5833b4c
·
1 Parent(s): c3f782c

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - zh
4
+ library_name: transformers
5
+ pipeline_tag: text2text-generation
6
+ ---
7
+
8
+ ```python
9
+ from transformers import AutoTokenizer, AutoModelForCausalLM
10
+
11
+ tokenizer = AutoTokenizer.from_pretrained("svjack/bloom-daliy-dialogue")
12
+
13
+ model = AutoModelForCausalLM.from_pretrained("svjack/bloom-daliy-dialogue")
14
+
15
+ tokenizer.decode(
16
+ model.generate(
17
+ tokenizer.encode(
18
+ "你饿吗?", return_tensors="pt", add_special_tokens=True
19
+ ), 128)[0],
20
+ skip_special_tokens = True
21
+ ).split("\n-----\n")
22
+
23
+ '''
24
+ ['你饿吗?',
25
+ '我饿了',
26
+ '你想吃点什么吗?',
27
+ '我可能也饿了',
28
+ '你想吃块蛋糕吗?',
29
+ '这听起来不错',
30
+ '你要什么蛋糕?',
31
+ '我也不知道啊 第一次来这里',
32
+ '让我看看... 这是什么蛋糕?',
33
+ '是软软糖圣菲利斯蛋糕。',
34
+ '你吃了多少?',
35
+ '我大概应该吃一小百二十块。',
36
+ '你通常吃哪种蛋糕?',
37
+ '我通常只吃一点。',
38
+ '你知道,我不喜欢吃软糖圣菲利斯蛋糕。',
39
+ '你有什么特别喜欢']
40
+ '''
41
+ ```