Skip to content

Commit

Permalink
Add ERROR logging prefix and sort the prefixes alphabetically (#1308)
Browse files Browse the repository at this point in the history
* Add ERROR logging prefix and sort alphabetically

* fix comment
  • Loading branch information
TheBatmanofButler authored Oct 17, 2024
1 parent 3272032 commit 59a5236
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions megatron/neox_arguments/arguments.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,16 +50,19 @@
ATTENTION_TYPE_CHOICES,
)

### Logging colors ###
### ANSI escape codes ###
END = "\033[0m"
GREEN = "\033[92m"
RED = "\033[91m"
YELLOW = "\033[93m"
END = "\033[0m"
SUCCESS = f"{GREEN} [SUCCESS] {END}"
OKAY = f"{GREEN}[OKAY]{END}"
WARNING = f"{YELLOW}[WARNING]{END}"

### Formatted logging prefixes ###
ERROR = f"{RED}[ERROR]{END} "
FAIL = f"{RED}[FAIL]{END}"
INFO = "[INFO]"
OKAY = f"{GREEN}[OKAY]{END}"
SUCCESS = f"{GREEN} [SUCCESS] {END}"
WARNING = f"{YELLOW}[WARNING]{END}"

# ZERO defaults by deespeed
# These values should not be changed unless defaults in deepspeed are changed
Expand Down

0 comments on commit 59a5236

Please sign in to comment.