logging improvements (#1808) [skip ci]

* logging improvements * fix sort
2024-08-06 10:31:50 -04:00
parent 35d5e59d78
commit c56e0a79a5
3 changed files with 10 additions and 5 deletions
--- a/src/axolotl/utils/models.py
+++ b/src/axolotl/utils/models.py
@@ -1022,7 +1022,7 @@ def load_lora(model, cfg, inference=False, config_only=False):

    if cfg.lora_target_linear:
        linear_names = find_all_linear_names(model)
-        LOG.info(f"found linear modules: {repr(linear_names)}")
+        LOG.info(f"found linear modules: {repr(sorted(linear_names))}")
        lora_target_modules = list(set(lora_target_modules + linear_names))

    lora_config_kwargs = {}