Update modeling_tinylm.py

modeling_tinylm.py

@@ -114,4 +114,4 @@ def generate(model, tokenizer, prompt, max_new_tokens=100, temperature=0.8, top_
 if __name__ == "__main__":
     model, tokenizer, config = load_tinylm("./tinylm")
     print("Model loaded!")
-    print(generate(model, tokenizer, "Once upon a time"))
+    print("Use 'generate(model, tokenizer, \"Once upon a time, \"' to generate."))