update axolotl ascii art

axolotl-ai-cloud · Oct 10, 2024 · 4589f97 · 4589f97
1 parent 8159cbd
commit 4589f97
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py
@@ -55,8 +55,22 @@
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 
-
-def print_axolotl_text_art(suffix=None):
+AXOLOTL_LOGO = """
+ #@@+#@@      #@#+#@#
+@@. @@.         @@  @@           =@@#                               @@.                #@=   =@@#.
+@@    #@@@@@@@@@    @@           #@#@=                              @@.                #@=   ..=@@
+  #@@@@@@@@@@@@@@@@@            =@#.@#    .##=    .##+  .=####=+    @@.    .=#####+  =#@@###.  +@@
+@@@@@@@@@@+  +@@+  +@@          #@. =@=    .#@=  +@@+  =@#+..+#@#   @@.   =@#+..+#@# ..#@=..   +@@
+@@@@@@@@@@  ##@@  ##@@         =@#  .@#      =@#=@#.   @@.     @@.  @@.   @@.     #@+  #@=     +@@
+=@@@@@@@@@@@@@@@@@@@@=         #@=+++#@=      =@@#     @@.     @@.  @@.   @@.     #@+  #@=     +@@
+                              =@#=====@@     =@#=@#.   @@.     @@.  @@.   @@.     #@+  #@=     +@@
+@@@@@@@@@@@@@@@@  @@@@        #@.     #@=  .#@=  +@@+  #@#.  .=@#   @@+.. =@#.  .=@#   #@=..   +@@
+                             =@#      .@# .#@=    .#@=  =#@@@@#=    +#@@=  +#@@@@#=    .##@@+  +@@
+@@@@  @@@@@@@@@@@@@@@@
+"""
+
+
+def print_legacy_axolotl_text_art(suffix=None):
     font = "nancyj"
     ascii_text = "  axolotl"
     if suffix:
@@ -69,6 +83,13 @@ def print_axolotl_text_art(suffix=None):
     print_dep_versions()
 
 
+def print_axolotl_text_art(
+    **kwargs,  # pylint: disable=unused-argument
+):
+    if is_main_process():
+        print(AXOLOTL_LOGO)
+
+
 def print_dep_versions():
     packages = ["accelerate", "peft", "transformers", "trl", "torch", "bitsandbytes"]
     max_len = max(len(pkg) for pkg in packages)