Skip to content

Commit 1342063

Browse files
authored
fix lm_eval import [3x](#1708)
Signed-off-by: Kaihui-intel <[email protected]>
1 parent ce9b164 commit 1342063

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/llm/run_clm_no_trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,7 @@ def run_fn_for_gptq(model, dataloader_for_calibration, *args):
337337

338338
if args.accuracy:
339339
user_model.eval()
340-
from intel_extension_for_transformers.llm.evaluation.lm_eval import evaluate
340+
from intel_extension_for_transformers.transformers.llm.evaluation.lm_eval import evaluate
341341

342342
results = evaluate(
343343
model="hf-causal",
@@ -360,7 +360,7 @@ def run_fn_for_gptq(model, dataloader_for_calibration, *args):
360360

361361
if args.performance:
362362
user_model.eval()
363-
from intel_extension_for_transformers.llm.evaluation.lm_eval import evaluate
363+
from intel_extension_for_transformers.transformers.llm.evaluation.lm_eval import evaluate
364364
import time
365365

366366
samples = args.iters * args.batch_size

0 commit comments

Comments
 (0)