better dtype handle in loading

Former-commit-id: 663f0577dd61a1a31191db2c6fbb0c7cea533b21
2024-05-17 02:14:56 +08:00
parent a3320f26cf
commit 969e605c7e
3 changed files with 15 additions and 8 deletions
--- a/src/llamafactory/model/loader.py
+++ b/src/llamafactory/model/loader.py
@@ -170,6 +170,7 @@ def load_model(
        )
    else:
        param_stats = "all params: {:d}".format(all_param)
+
    logger.info(param_stats)

    if model_args.print_param_status: