Update README.md
Browse files
README.md
CHANGED
@@ -18,10 +18,8 @@ Install the runtime library first:
|
|
18 |
```bash
|
19 |
pip install transformers
|
20 |
```
|
21 |
-
To
|
22 |
```python
|
23 |
-
|
24 |
-
|
25 |
import torch
|
26 |
from transformers import PreTrainedTokenizerFast
|
27 |
from torch.cuda.amp import autocast
|
@@ -53,12 +51,9 @@ for i in range(0, num_texts, batch_size):
|
|
53 |
for output_sequence in output:
|
54 |
generated_text = tokenizer.decode(output_sequence, skip_special_tokens=True)
|
55 |
print(generated_text)
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
```
|
60 |
|
61 |
|
62 |
|
63 |
#### Hardware
|
64 |
-
Model was trained for
|
|
|
18 |
```bash
|
19 |
pip install transformers
|
20 |
```
|
21 |
+
To generate a new gene sequence using the model:
|
22 |
```python
|
|
|
|
|
23 |
import torch
|
24 |
from transformers import PreTrainedTokenizerFast
|
25 |
from torch.cuda.amp import autocast
|
|
|
51 |
for output_sequence in output:
|
52 |
generated_text = tokenizer.decode(output_sequence, skip_special_tokens=True)
|
53 |
print(generated_text)
|
|
|
|
|
|
|
54 |
```
|
55 |
|
56 |
|
57 |
|
58 |
#### Hardware
|
59 |
+
Model was trained for 8 hours on 4 Nvidia A100-40G GPUs.
|