segestic
/

Tinystories-gpt-0.1-3m

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

segestic commited on Dec 7, 2023

Commit

827dae5

•

1 Parent(s): 3ae7ea1

Update README.md

Files changed (1) hide show

README.md +3 -0

README.md CHANGED Viewed

@@ -15,8 +15,11 @@ pipeline_tag: text-generation
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
 model = AutoModelForCausalLM.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
 prompt = "Once upon a time there was"
 input_ids = tokenizer.encode(prompt, return_tensors="pt")
 #### Generate completion
 output = model.generate(input_ids, max_length = 1000, num_beams=1)

 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
 model = AutoModelForCausalLM.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
 prompt = "Once upon a time there was"
 input_ids = tokenizer.encode(prompt, return_tensors="pt")
 #### Generate completion
 output = model.generate(input_ids, max_length = 1000, num_beams=1)