laituan245 commited on
Commit
e69fcaf
·
1 Parent(s): fe6e05d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -5
README.md CHANGED
@@ -5,12 +5,7 @@ license: apache-2.0
5
  ```python
6
  from transformers import AutoTokenizer, T5ForConditionalGeneration
7
  tokenizer = AutoTokenizer.from_pretrained("laituan245/molt5-small", model_max_length=512)
8
- tokenizer.save_pretrained('tokenizer')
9
  model = T5ForConditionalGeneration.from_pretrained('laituan245/molt5-small')
10
- input_text = 'CC1=CC=CC=<extra_id_0>'
11
- input_ids = tokenizer(input_text, return_tensors="pt").input_ids
12
- outputs = model.generate(input_ids, num_beams=5, max_length=512)
13
- print(tokenizer.decode(outputs[0], skip_special_tokens=False))
14
  ```
15
 
16
  ## Paper
 
5
  ```python
6
  from transformers import AutoTokenizer, T5ForConditionalGeneration
7
  tokenizer = AutoTokenizer.from_pretrained("laituan245/molt5-small", model_max_length=512)
 
8
  model = T5ForConditionalGeneration.from_pretrained('laituan245/molt5-small')
 
 
 
 
9
  ```
10
 
11
  ## Paper