tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

log rank too (#527)

Browse Source
This commit is contained in:
Wing Lian
2023-09-06 08:37:51 -04:00
committed by GitHub
parent a546ca2813
commit 245c5c41e2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/logging_config.py
View File

@@ -23,6 +23,7 @@ class ColorfulFormatter(Formatter):
} }
def format(self, record): def format(self, record):
record.rank = int(os.getenv("LOCAL_RANK", "0"))
log_message = super().format(record) log_message = super().format(record)
return self.COLORS.get(record.levelname, "") + log_message + Fore.RESET return self.COLORS.get(record.levelname, "") + log_message + Fore.RESET
@@ -35,7 +36,7 @@ DEFAULT_LOGGING_CONFIG: Dict[str, Any] = {
}, },
"colorful": { "colorful": {
"()": ColorfulFormatter, "()": ColorfulFormatter,
"format": "[%(asctime)s] [%(levelname)s] [%(name)s.%(funcName)s:%(lineno)d] [PID:%(process)d] %(message)s", "format": "[%(asctime)s] [%(levelname)s] [%(name)s.%(funcName)s:%(lineno)d] [PID:%(process)d] [RANK:%(rank)d] %(message)s",
}, },
}, },
"filters": {}, "filters": {},