better dtype handle in loading

2024-05-17 02:14:56 +08:00
parent ddec9e1b84
commit d9f190ff1e
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: