diff --git a/pyproject.toml b/pyproject.toml
index 8bb082d..6af9563 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llm_atc"
-version = "0.1.5"
+version = "0.1.6"
 description = "Tools for fine tuning and serving LLMs"
 authors = ["Andrew Aikawa <asai@berkeley.edu>"]
 readme = "README.md"