hu-lab commited on
Commit
e081ac4
·
verified ·
1 Parent(s): d84c46c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -7
README.md CHANGED
@@ -18,10 +18,8 @@ Install the runtime library first:
18
  ```bash
19
  pip install transformers
20
  ```
21
- To calculate the embedding of a dna sequence:
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 468 hours on 8 Nvidia A800-80G GPUs.
 
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.