Update README.md
Browse files
README.md
CHANGED
@@ -15,8 +15,11 @@ pipeline_tag: text-generation
|
|
15 |
|
16 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
17 |
tokenizer = AutoTokenizer.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
|
|
|
18 |
model = AutoModelForCausalLM.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
|
|
|
19 |
prompt = "Once upon a time there was"
|
|
|
20 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
21 |
#### Generate completion
|
22 |
output = model.generate(input_ids, max_length = 1000, num_beams=1)
|
|
|
15 |
|
16 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
17 |
tokenizer = AutoTokenizer.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
|
18 |
+
|
19 |
model = AutoModelForCausalLM.from_pretrained("segestic/Tinystories-gpt-0.1-3m")
|
20 |
+
|
21 |
prompt = "Once upon a time there was"
|
22 |
+
|
23 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
24 |
#### Generate completion
|
25 |
output = model.generate(input_ids, max_length = 1000, num_beams=1)
|