diff --git a/pyproject.toml b/pyproject.toml
index 77758a9d..c9c26880 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ readme = "README.md"
 license = { file = "LICENSE" }
 authors = [{ name = "Replicate, Inc." }]
 requires-python = ">=3.8"
-dependencies = ["packaging", "pydantic>1,<2", "requests>2"]
+dependencies = ["packaging", "pydantic>1", "requests>2"]
 optional-dependencies = { dev = [
     "black",
     "mypy",
diff --git a/replicate/base_model.py b/replicate/base_model.py
index b3cf1d48..c1dc1498 100644
--- a/replicate/base_model.py
+++ b/replicate/base_model.py
@@ -4,7 +4,10 @@
     from replicate.client import Client
     from replicate.collection import Collection
 
-import pydantic
+try:
+    from pydantic import v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
 
 
 class BaseModel(pydantic.BaseModel):
diff --git a/requirements.txt b/requirements.txt
index 0661ab73..85b65de2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,7 +12,7 @@ idna==3.4
     # via requests
 packaging==23.0
     # via replicate (pyproject.toml)
-pydantic==1.10.7
+pydantic>=1.10.7
     # via replicate (pyproject.toml)
 requests==2.31.0
     # via replicate (pyproject.toml)