broskicodes
/

simple-stories-4M

Text Generation

simple_stories_4m

text-generation-inference

Model card Files Files and versions Community

broskicodes commited on Jan 8

Commit

c5b6d17

•

1 Parent(s): 1724f13

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ tokenizer = AutoTokenizer.from_pretrained("broskicodes/simple-stories-4M", trust
 ```
 inputs = tokenizer("Once upon a time,", return_tensors="pt", return_attention_mask=False)
-outputs = model.model.generate(**inputs, 250)
 ```
 Note that we are calling `model.model.generate` not just `model.generate`

 ```
 inputs = tokenizer("Once upon a time,", return_tensors="pt", return_attention_mask=False)
+outputs = model.model.generate(inputs['input_ids'], 250)
 ```
 Note that we are calling `model.model.generate` not just `model.generate`