Skip to content

Commit

Permalink
Fix text-generation example accuracy test (#1262)
Browse files Browse the repository at this point in the history
  • Loading branch information
changwangss committed Feb 6, 2024
1 parent 85c9c9c commit a2cfb80
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -285,7 +285,7 @@
scale_dtype=args.woq_scale_dtype,
weight_dtype=args.woq_weight_dtype,
scheme=args.woq_scheme,
group_size=args.gptq_block_size,
group_size=args.woq_group_size,
algorithm=args.woq_algo,
tokenizer=tokenizer,
algorithm_args=algorithm_args,
Expand Down Expand Up @@ -438,7 +438,7 @@
peft_config.base_model_name_or_path if args.peft_model_id else args.model
)
from intel_extension_for_transformers.llm.evaluation.lm_eval import evaluate

args._commit_hash = "main" if args._commit_hash is None else args._commit_hash
results = evaluate(
model="hf-causal",
model_args="pretrained="
Expand Down

0 comments on commit a2cfb80

Please sign in to comment.