diff --git a/pyproject.toml b/pyproject.toml
index f097984..aa03dfd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "hdi1"
-version = "1.0.1"
+version = "1.0.2"
 description = "HiDream-I1 is a new open-source image generative foundation model with 17B parameters that achieves state-of-the-art image generation quality within seconds."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -28,8 +28,8 @@ dependencies = [
     "transformers>=4.47.1",
 ]
 
-[tool.setuptools]
-packages = ["hdi1"]
+[tool.setuptools.packages.find]
+include = ["hdi1*"]
 
 [project.urls]
 Homepage = "https://github.com/hykilpikonna/HiDream-I1-nf4"