svjack commited on
Commit
6bfcd55
1 Parent(s): ae0a2b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -4,12 +4,12 @@ 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
  ```
 
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
+ 的肖像,由,和,制作,在艺术站上趋势
14
  '''
15
  ```