Update README.md

README.md

@@ -45,13 +45,16 @@ system_prompt = """<|SYSTEM|># StableLM Tuned (Alpha version)
 - StableLM will refuse to participate in anything that could harm a human.
 """
 
-prompt = f"{system_prompt}<|USER|>What's your mood today<|ASSISTANT|>?"
+prompt = f"{system_prompt}<|USER|>What's your mood today?<|ASSISTANT|>"
 
 inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
 tokens = model.generate(
   **inputs,
   max_new_tokens=64,
-  StoppingCriteriaList([StopOnTokens()]))
+  temperature=0.7,
+  do_sample=True,
+  StoppingCriteriaList([StopOnTokens()])
+)
 print(tokenizer.decode(tokens[0], skip_special_tokens=True))
 ```