GanymedeNil commited on
Commit
06a55a1
1 Parent(s): 166e592

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -33,7 +33,7 @@ Title:
33
  """ # 固定格式
34
  encoding = tokenizer(input_text, return_tensors="pt").to("cuda")
35
 
36
- outputs = model.generate(**encoding,max_length=8000,temperature=0.2,do_sample=True)
37
  generated_ids = outputs[:, encoding.input_ids.shape[1]:]
38
  generated_texts = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
39
  print(generated_texts[0])
 
33
  """ # 固定格式
34
  encoding = tokenizer(input_text, return_tensors="pt").to("cuda")
35
 
36
+ outputs = model.generate(**encoding,max_length=8192,temperature=0.2,do_sample=True)
37
  generated_ids = outputs[:, encoding.input_ids.shape[1]:]
38
  generated_texts = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
39
  print(generated_texts[0])