Skip to content

Commit fbb9c13

Browse files
authored
update gguf and support for CompressedLinear (#950)
1 parent 824a21f commit fbb9c13

File tree

5 files changed

+1336
-482
lines changed

5 files changed

+1336
-482
lines changed

auto_round/__main__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -789,7 +789,7 @@ def run_eval():
789789
if args.eval_task_by_task:
790790
eval_task_by_task(
791791
model=args.model,
792-
device=args.device,
792+
device=args.device_map,
793793
tasks=args.tasks,
794794
batch_size=args.eval_bs,
795795
trust_remote_code=not args.disable_trust_remote_code,

auto_round/export/export_to_gguf/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -412,7 +412,7 @@ def prepare_tensors(cls):
412412
skip = False
413413
for tensor_info in cls.gguf_writer.tensors:
414414
if new_name in tensor_info:
415-
print("new_name already add to gguf_writer, skip")
415+
logger.warning(f"{new_name} already add to gguf_writer, skip")
416416
skip = True
417417
break
418418
if skip:

0 commit comments

Comments
 (0)