diff --git a/adetailer/args.py b/adetailer/args.py
index bda899d..5dd4577 100644
--- a/adetailer/args.py
+++ b/adetailer/args.py
@@ -5,16 +5,28 @@ from dataclasses import dataclass
 from functools import cached_property, partial
 from typing import Any, Literal, NamedTuple, Optional
 
-from pydantic import (
-    BaseModel,
-    Extra,
-    NonNegativeFloat,
-    NonNegativeInt,
-    PositiveInt,
-    confloat,
-    conint,
-    validator,
-)
+try:
+    from pydantic.v1 import (
+        BaseModel,
+        Extra,
+        NonNegativeFloat,
+        NonNegativeInt,
+        PositiveInt,
+        confloat,
+        conint,
+        validator,
+    )
+except ImportError:
+    from pydantic import (
+        BaseModel,
+        Extra,
+        NonNegativeFloat,
+        NonNegativeInt,
+        PositiveInt,
+        confloat,
+        conint,
+        validator,
+    )
 
 
 @dataclass
diff --git a/pyproject.toml b/pyproject.toml
index 410827b..9b0822f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ license = { text = "AGPL-3.0" }
 dependencies = [
     "ultralytics>=8.1",
     "mediapipe>=10",
-    "pydantic<2",
+    "pydantic<3",
     "rich>=13",
     "huggingface_hub",
 ]