From 3ac83f8396be91d63a34e01dd34549667277bfa7 Mon Sep 17 00:00:00 2001 From: "Chu,Youcheng" <70999398+cranechu0131@users.noreply.github.com> Date: Thu, 15 Aug 2024 13:40:01 +0800 Subject: [PATCH] fix: delete ipex extension import in ppl wikitext evaluation (#11806) Co-authored-by: Jinhe Tang --- python/llm/dev/benchmark/perplexity/run_wikitext.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/python/llm/dev/benchmark/perplexity/run_wikitext.py b/python/llm/dev/benchmark/perplexity/run_wikitext.py index 190d5114a86..58e5f587386 100644 --- a/python/llm/dev/benchmark/perplexity/run_wikitext.py +++ b/python/llm/dev/benchmark/perplexity/run_wikitext.py @@ -38,8 +38,6 @@ if args.precision == "fp16": # ipex fp16 from transformers import AutoModelForCausalLM - if "xpu" in args.device: - import intel_extension_for_pytorch as ipex model = AutoModelForCausalLM.from_pretrained(args.model_path, use_cache=args.use_cache, trust_remote_code=True) model = model.half() else: # ipex-llm