diff --git a/pyproject.toml b/pyproject.toml
index 5da8c13..4a43b66 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "comfy-modelopt"
-version = "0.2.1"
+version = "0.2.2"
 description = "NVIDIA ModelOpt integration for ComfyUI - Quantize and optimize diffusion models with INT8/FP8/INT4"
 readme = "README.md"
 requires-python = ">=3.10,<3.13"
@@ -74,7 +74,7 @@ target-version = ['py310', 'py311', 'py312']
 display_name = "NVIDIA ModelOpt"
 description = "Quantize and optimize diffusion models with NVIDIA ModelOpt (INT8/FP8/INT4)"
 author = "ComfyUI ModelOpt Contributors"
-version = "0.2.1"
+version = "0.2.2"
 node_list = [
     "ModelOptUNetLoader",
     "ModelOptQuantizeUNet",