diff --git a/pyproject.toml b/pyproject.toml
index 3796d11..ded5f9f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "granite-switch"
-version = "0.1.0"
+version = "0.1.1"
 description = "Granite Switch: Composable model building"
 readme = "README.md"
 license = "Apache-2.0"
diff --git a/uv.lock b/uv.lock
index 71c7739..5ae4471 100644
--- a/uv.lock
+++ b/uv.lock
@@ -2440,7 +2440,7 @@ wheels = [
 
 [[package]]
 name = "granite-switch"
-version = "0.1.0"
+version = "0.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "torch", version = "2.10.0", source = { registry = "https://pypi.org/simple" }, marker = "extra == 'extra-14-granite-switch-tutorials' or extra == 'extra-14-granite-switch-vllm' or extra == 'group-14-granite-switch-dev' or extra == 'group-14-granite-switch-test' or extra == 'group-14-granite-switch-vllm19'" },