UniverseTBD
/

falcon-7b-abstracts-tiny

Text Generation

RefinedWebModel

text-generation-inference

Model card Files Files and versions

charlieoneill commited on Jul 20, 2023

Commit

37cbdec

·

1 Parent(s): 6b05ab1

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -13,6 +13,7 @@ It was fine-tuned on several thousand astronomy abstracts collected on Arxiv.
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 online_model = AutoModelForCausalLM.from_pretrained("universeTBD/falcon-7b-abstracts-tiny", torch_dtype=torch.bfloat16,

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import transformers
 import torch
 online_model = AutoModelForCausalLM.from_pretrained("universeTBD/falcon-7b-abstracts-tiny", torch_dtype=torch.bfloat16,