From 3699d1d3baf89a7d78a7204f78df25ed2dc34f44 Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 23 Jun 2025 15:20:56 -0400 Subject: [PATCH] hotfix llama3 temperature is float (#10938) --- examples/llama3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/llama3.py b/examples/llama3.py index 2628e3dcff..9664f491f2 100644 --- a/examples/llama3.py +++ b/examples/llama3.py @@ -248,7 +248,7 @@ if __name__ == "__main__": parser.add_argument("--port", type=int, default=7776, help="Web server port") parser.add_argument("--debug", action="store_true", help="Enable debug mode") parser.add_argument("--seed", type=int, help="Random seed") - parser.add_argument("--temperature", type=int, default=0.85, help="Temperature") + parser.add_argument("--temperature", type=float, default=0.85, help="Temperature") parser.add_argument("--benchmark", action="store_true", help="Run a benchmark") parser.add_argument("--timing", action="store_true", help="Print timing per token") parser.add_argument("--profile", action="store_true", help="Output profile data")