marsggbo commited on
Commit
0678573
1 Parent(s): d9b71eb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -6,6 +6,8 @@ base_model:
6
 
7
  ```python
8
  import os
 
 
9
  from glob import glob
10
  from transformers import AutoModelForSeq2SeqLM, AutoConfig
11
 
 
6
 
7
  ```python
8
  import os
9
+ import torch
10
+
11
  from glob import glob
12
  from transformers import AutoModelForSeq2SeqLM, AutoConfig
13