UniverseTBD
/

astrollama

Text Generation

text-generation-inference

Model card Files Files and versions Community

joshnguyen commited on Aug 26, 2023

Commit

c4f4220

•

1 Parent(s): 6c1920c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ from transformers import pipeline
 generator = pipeline(
     task="text-generation",
     model=model,
-    tokenizer="tokenizer",
     device_map="auto"
 )

 generator = pipeline(
     task="text-generation",
     model=model,
+    tokenizer=tokenizer,
     device_map="auto"
 )