Commit
·
37cbdec
1
Parent(s):
6b05ab1
Update README.md
Browse files
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,
|