@@ -1479,7 +1479,7 @@ def main(args_in: list[str] | None = None) -> None:
1479
1479
# Avoid printing anything besides the dump output
1480
1480
logging .basicConfig (level = logging .CRITICAL )
1481
1481
else :
1482
- logging .basicConfig (level = logging .INFO )
1482
+ logging .basicConfig (level = logger .INFO )
1483
1483
1484
1484
if args .no_vocab and args .vocab_only :
1485
1485
raise ValueError ("--vocab-only does not make sense with --no-vocab" )
@@ -1520,7 +1520,7 @@ def main(args_in: list[str] | None = None) -> None:
1520
1520
"q8_0" : GGMLFileType .MostlyQ8_0 ,
1521
1521
}[args .outtype ]
1522
1522
1523
- logging .info (f"params = { params } " )
1523
+ logger .info (f"params = { params } " )
1524
1524
1525
1525
model_parent_path = model_plus .paths [0 ].parent
1526
1526
vocab_path = Path (args .vocab_dir or args .model or model_parent_path )
@@ -1535,26 +1535,26 @@ def main(args_in: list[str] | None = None) -> None:
1535
1535
outfile = args .outfile
1536
1536
OutputFile .write_vocab_only (outfile , params , vocab , special_vocab ,
1537
1537
endianess = endianess , pad_vocab = args .pad_vocab )
1538
- logging .info (f"Wrote { outfile } " )
1538
+ logger .info (f"Wrote { outfile } " )
1539
1539
return
1540
1540
1541
1541
if model_plus .vocab is not None and args .vocab_dir is None and not args .no_vocab :
1542
1542
vocab = model_plus .vocab
1543
1543
1544
- logging .info (f"Vocab info: { vocab } " )
1545
- logging .info (f"Special vocab info: { special_vocab } " )
1544
+ logger .info (f"Vocab info: { vocab } " )
1545
+ logger .info (f"Special vocab info: { special_vocab } " )
1546
1546
model = model_plus .model
1547
1547
model = convert_model_names (model , params , args .skip_unknown )
1548
1548
ftype = pick_output_type (model , args .outtype )
1549
1549
model = convert_to_output_type (model , ftype )
1550
1550
outfile = args .outfile or default_outfile (model_plus .paths , ftype )
1551
1551
1552
1552
params .ftype = ftype
1553
- logging .info (f"Writing { outfile } , format { ftype } " )
1553
+ logger .info (f"Writing { outfile } , format { ftype } " )
1554
1554
1555
1555
OutputFile .write_all (outfile , ftype , params , model , vocab , special_vocab ,
1556
1556
concurrency = args .concurrency , endianess = endianess , pad_vocab = args .pad_vocab )
1557
- logging .info (f"Wrote { outfile } " )
1557
+ logger .info (f"Wrote { outfile } " )
1558
1558
1559
1559
1560
1560
if __name__ == '__main__' :
0 commit comments