svjack commited on
Commit
ae0a2b1
1 Parent(s): e601080

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ```python
2
+ from transformers import T5Tokenizer, MT5ForConditionalGeneration
3
+ model = "svjack/prompt-extend-chinese"
4
+ device = "cpu"
5
+ tokenizer = T5Tokenizer.from_pretrained(model)
6
+ model = MT5ForConditionalGeneration.from_pretrained(model).to(device).eval()
7
+ prompt = "卡通龙"
8
+ encode = tokenizer(prompt, return_tensors='pt').to(device)
9
+ answer = model.generate(encode.input_ids)[0]
10
+ decoded = tokenizer.decode(answer, skip_special_tokens=True)
11
+ decoded
12
+ '''
13
+ 数字艺术、艺术品趋势、电影照明、工作室质量、光滑成型、5
14
+ '''
15
+ ```