fix: delete ipex extension import in ppl wikitext evaluation (intel-a…

…nalytics#11806) Co-authored-by: Jinhe Tang <[email protected]>
hxsz1997 · Aug 15, 2024 · 3ac83f8 · 3ac83f8
1 parent 016e840
commit 3ac83f8
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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