diff --git a/riffusion/py.typed b/riffusion/py.typed
new file mode 100644
index 0000000..82e5f80
--- /dev/null
+++ b/riffusion/py.typed
@@ -0,0 +1 @@
+# https://peps.python.org/pep-0561/
diff --git a/setup.py b/setup.py
index c33d7d5..fdc7b79 100644
--- a/setup.py
+++ b/setup.py
@@ -28,6 +28,9 @@ setup(
         "transformers",
     ],
     install_requires=packages,
+    package_data={
+        "riffusion": ["py.typed"],
+    },
     classifiers=[
         "Development Status :: 4 - Beta",
         "Intended Audience :: Developers",