diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py index db975501a..443f0322c 100644 --- 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)