diff --git a/pyproject.toml b/pyproject.toml
index 3a60576..de49b27 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llm-analysis"
-version = "0.2.1"
+version = "0.2.2"
 description = "Latency and Memory Analysis of Transformer Models for Training and Inference"
 authors = ["Cheng Li <pistasable@gmail.com>"]
 license = "Apache License, Version 2.0"