From eda17e9452f41e4bfd3f2e2aba9689c90afc58c0 Mon Sep 17 00:00:00 2001 From: Jinhe Tang Date: Thu, 15 Aug 2024 13:21:33 +0800 Subject: [PATCH] fix: delete ipex extension import in ppl wikitext evaluation --- 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