universeTBD
/

falcon-7b-abstracts-tiny

Text Generation

RefinedWebModel

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

charlieoneill commited on Jul 19, 2023

Commit

71f9155

•

1 Parent(s): 1a3ba52

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ It was fine-tuned on several thousand astronomy abstracts collected on Arxiv.
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-online_model = AutoModelForCausalLM.from_pretrained("charlieoneill/falcon-7b-abstracts-2190", torch_dtype=torch.bfloat16,
                                                     device_map="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b")

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+online_model = AutoModelForCausalLM.from_pretrained("universeTBD/falcon-7b-abstracts-tiny", torch_dtype=torch.bfloat16,
                                                     device_map="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b")