OpenGenerativeAI
/

testModel

Text Generation

text-generation-inference

Model card Files Files and versions

futureHQ commited on Mar 20, 2024

Commit

f3c59b9

·

verified ·

1 Parent(s): ffceb72

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ You will need the transformers>=4.31
 from transformers import AutoTokenizer
 import transformers
 import torch
-model = "PY007/TinyLlama-1.1B-intermediate-step-240k-503b"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",

 from transformers import AutoTokenizer
 import transformers
 import torch
+model = "OpenGenerativeAI/testModel"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",