seonghyeonye commited on
Commit
7be61e6
·
1 Parent(s): fe01255

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,8 +19,8 @@ Here is how to download the model in PyTorch:
19
  import torch
20
  from transformers import T5Tokenizer, T5ForConditionalGeneration
21
 
22
- model = T5ForConditionalGeneration.from_pretrained("seonghyeonye/flipped_11B")
23
- tokenizer = T5Tokenizer.from_pretrained("seonghyeonye/flipped_11B")
24
  ```
25
  If you want to use another checkpoint, please replace the path in `T5Tokenizer` and `T5ForConditionalGeneration`.
26
  We also provide a quick [Jupyter Notebook](https://github.com/seonghyeonye/Flipped-Learning/blob/master/flipped_inference.ipynb) where you can inference with our method.
 
19
  import torch
20
  from transformers import T5Tokenizer, T5ForConditionalGeneration
21
 
22
+ model = T5ForConditionalGeneration.from_pretrained("seonghyeonye/flipped_3B")
23
+ tokenizer = T5Tokenizer.from_pretrained("seonghyeonye/flipped_3B")
24
  ```
25
  If you want to use another checkpoint, please replace the path in `T5Tokenizer` and `T5ForConditionalGeneration`.
26
  We also provide a quick [Jupyter Notebook](https://github.com/seonghyeonye/Flipped-Learning/blob/master/flipped_inference.ipynb) where you can inference with our method.