stabilityai
/

stablelm-tuned-alpha-7b

Text Generation

text-generation-inference

Model card Files Files and versions

dmayhem93 commited on Apr 19, 2023

Commit

e736571

·

1 Parent(s): 2fc0ec9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -53,7 +53,7 @@ tokens = model.generate(
   max_new_tokens=64,
   temperature=0.7,
   do_sample=True,
-  StoppingCriteriaList([StopOnTokens()])
 )
 print(tokenizer.decode(tokens[0], skip_special_tokens=True))
 ```

   max_new_tokens=64,
   temperature=0.7,
   do_sample=True,
+  stopping_criteria=StoppingCriteriaList([StopOnTokens()])
 )
 print(tokenizer.decode(tokens[0], skip_special_tokens=True))
 ```