charlieoneill commited on
Commit
37cbdec
1 Parent(s): 6b05ab1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -13,6 +13,7 @@ It was fine-tuned on several thousand astronomy abstracts collected on Arxiv.
13
 
14
  ```python
15
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
16
  import torch
17
 
18
  online_model = AutoModelForCausalLM.from_pretrained("universeTBD/falcon-7b-abstracts-tiny", torch_dtype=torch.bfloat16,
 
13
 
14
  ```python
15
  from transformers import AutoModelForCausalLM, AutoTokenizer
16
+ import transformers
17
  import torch
18
 
19
  online_model = AutoModelForCausalLM.from_pretrained("universeTBD/falcon-7b-abstracts-tiny", torch_dtype=torch.bfloat16,