diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py index db975501a3..fd5ab3e56c 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)