Skip to content

Commit c729e03

Browse files
add log (#2275)
Signed-off-by: ZX-ModelCloud <zx@modelcloud.ai>
1 parent f545879 commit c729e03

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

gptqmodel/utils/importer.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,17 +370,19 @@ def select_quant_linear(
370370
#if not message_logged:
371371
# logger.info(f"Auto pick kernel based on compatibility: {cls}")
372372
# message_logged = True
373-
log.info(f"{'Packing' if pack else ''} Kernel: Auto-selection: adding candidate `{cls.__name__}`")
373+
log.info(f"{'Packing ' if pack else ''}Kernel: Auto-selection: adding candidate `{cls.__name__}`")
374374
validated_qlinears.append(cls)
375375
if not multi_select:
376+
log.info(f"Kernel: selected -> `{cls.__name__}`.")
376377
return cls
377378
else:
378379
#if not message_logged:
379380
# logger.info(f"Auto pick kernel based on compatibility: {cls}")
380381
# message_logged = True
381-
log.info(f"{'Packing' if pack else ''} Kernel: Auto-selection: adding candidate `{cls.__name__}`")
382+
log.info(f"{'Packing ' if pack else ''}Kernel: Auto-selection: adding candidate `{cls.__name__}`")
382383
validated_qlinears.append(cls)
383384
if not multi_select:
385+
log.info(f"Kernel: selected -> `{cls.__name__}`.")
384386
return cls
385387

386388
if err:

0 commit comments

Comments
 (0)