support commit info

Former-commit-id: af752b1c27
2025-12-17 20:30:36 +08:00 · 2025-03-11 15:13:59 +08:00
parent 317d0855d2
commit 99b71768a0
1 changed files with 9 additions and 0 deletions
--- a/src/llamafactory/extras/env.py
+++ b/src/llamafactory/extras/env.py
@@ -74,4 +74,13 @@ def print_env() -> None:
    except Exception:
        pass

+    try:
+        import subprocess
+
+        commit_info = subprocess.run(["git", "rev-parse", "HEAD"], capture_output=True, text=True, check=True)
+        commit_hash = commit_info.stdout.strip()
+        info["Git commit"] = commit_hash
+    except Exception:
+        pass
+
    print("\n" + "\n".join([f"- {key}: {value}" for key, value in info.items()]) + "\n")