Update README.md
Browse files
README.md
CHANGED
|
@@ -68,6 +68,7 @@ with torch.no_grad():
|
|
| 68 |
## For Sequence Completion:
|
| 69 |
```python
|
| 70 |
from transformers import T5ForConditionalGeneration, T5Tokenizer, T5EncoderModel
|
|
|
|
| 71 |
import torch
|
| 72 |
|
| 73 |
sequence = "MDTAYPREDTRAPTPSKAGAHTALTLGAPHPPPRDHLIWSVFSTLYLNLCCLGFLALAYSIKARDQKVVGDLEAARRFGSKAKCYNILAAMWTLVPPLLLLGLVVTGALHLARLAKDSAAFFSTKFDDADYD"
|
|
|
|
| 68 |
## For Sequence Completion:
|
| 69 |
```python
|
| 70 |
from transformers import T5ForConditionalGeneration, T5Tokenizer, T5EncoderModel
|
| 71 |
+
from transformers.generation import GenerationConfig
|
| 72 |
import torch
|
| 73 |
|
| 74 |
sequence = "MDTAYPREDTRAPTPSKAGAHTALTLGAPHPPPRDHLIWSVFSTLYLNLCCLGFLALAYSIKARDQKVVGDLEAARRFGSKAKCYNILAAMWTLVPPLLLLGLVVTGALHLARLAKDSAAFFSTKFDDADYD"
|