diff --git a/python/llm/dev/benchmark/perplexity/run_wikitext.py b/python/llm/dev/benchmark/perplexity/run_wikitext.py index a5b7dea3..531ffff5 100644 --- a/python/llm/dev/benchmark/perplexity/run_wikitext.py +++ b/python/llm/dev/benchmark/perplexity/run_wikitext.py @@ -44,6 +44,7 @@ else: # ipex-llm use_cache=args.use_cache, trust_remote_code=True) model = model.half() model = model.to(args.device) +model = model.eval() with open(args.data_path, "rb") as f: data = f.read()