We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3272032 commit 59a5236Copy full SHA for 59a5236
1 file changed
megatron/neox_arguments/arguments.py
@@ -50,16 +50,19 @@
50
ATTENTION_TYPE_CHOICES,
51
)
52
53
-### Logging colors ###
+### ANSI escape codes ###
54
+END = "\033[0m"
55
GREEN = "\033[92m"
56
RED = "\033[91m"
57
YELLOW = "\033[93m"
-END = "\033[0m"
58
-SUCCESS = f"{GREEN} [SUCCESS] {END}"
59
-OKAY = f"{GREEN}[OKAY]{END}"
60
-WARNING = f"{YELLOW}[WARNING]{END}"
+
+### Formatted logging prefixes ###
+ERROR = f"{RED}[ERROR]{END} "
61
FAIL = f"{RED}[FAIL]{END}"
62
INFO = "[INFO]"
63
+OKAY = f"{GREEN}[OKAY]{END}"
64
+SUCCESS = f"{GREEN} [SUCCESS] {END}"
65
+WARNING = f"{YELLOW}[WARNING]{END}"
66
67
# ZERO defaults by deespeed
68
# These values should not be changed unless defaults in deepspeed are changed
0 commit comments