From 440aab8a6fd2f233cec7a341d5f9cc0fe5c14a3a Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sat, 7 Dec 2024 22:23:33 -0500
Subject: [PATCH] add --version support to axolotl cli (#2152) [skip ci]

---
 src/axolotl/cli/main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/axolotl/cli/main.py b/src/axolotl/cli/main.py
index a138daab2..a77410776 100644
--- a/src/axolotl/cli/main.py
+++ b/src/axolotl/cli/main.py
@@ -5,6 +5,7 @@ from typing import Optional
 
 import click
 
+import axolotl
 from axolotl.cli.utils import (
     add_options_from_config,
     add_options_from_dataclass,
@@ -16,6 +17,7 @@ from axolotl.utils.config.models.input.v0_4_1 import AxolotlInputConfig
 
 
 @click.group()
+@click.version_option(version=axolotl.__version__, prog_name="axolotl")
 def cli():
     """Axolotl CLI - Train and fine-tune large language models"""