Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit d6e6e9f

Browse files
authored
fix typing (#1451)
1 parent d40be7f commit d6e6e9f

File tree

2 files changed

+1
-3
lines changed
  • examples/huggingface/pytorch/text-generation/quantization
  • intel_extension_for_transformers/transformers/llm/evaluation/lm_eval

2 files changed

+1
-3
lines changed

examples/huggingface/pytorch/text-generation/quantization/run_generation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,7 @@
541541
args._commit_hash = "main" if args._commit_hash is None else args._commit_hash
542542
results = evaluate(
543543
model="hf-causal",
544-
model_args="pretrained=" + argrs.model + ",tokenizer=" + args.model + ",dtype=float32" + ",_commit_hash=" + args._commit_hash +
544+
model_args="pretrained=" + args.model + ",tokenizer=" + args.model + ",dtype=float32" + ",_commit_hash=" + args._commit_hash +
545545
",trust_remote_code=" + str(args.trust_remote_code),
546546
user_model=user_model,
547547
batch_size=args.batch_size,

intel_extension_for_transformers/transformers/llm/evaluation/lm_eval/evaluator.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,6 @@ def evaluate(model,
124124
}
125125
if user_model:
126126
kwargs["init_empty_weights"] = True
127-
if "pretrained" not in model_args:
128-
model_args = "pretrained='Muennighoff/tiny-random-bert'," + model_args
129127

130128
if device == "hpu":
131129
# if hpu, set user_model

0 commit comments

Comments
 (0)