seems fine

lucidrains · Jul 15, 2023 · 2527ce3 · 2527ce3
1 parent c8b6778
commit 2527ce3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/imagen_pytorch/configs.py b/imagen_pytorch/configs.py
@@ -1,5 +1,5 @@
 import json
-from pydantic import BaseModel, validator, root_validator
+from pydantic import BaseModel, validator
 from typing import List, Iterable, Optional, Union, Tuple, Dict, Any
 from enum import Enum
 

diff --git a/imagen_pytorch/version.py b/imagen_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.25.4'
+__version__ = '1.25.5'
diff --git a/setup.py b/setup.py
@@ -38,7 +38,7 @@
     'numpy',
     'packaging',
     'pillow',
-    'pydantic',
+    'pydantic>=2',
     'pytorch-lightning',
     'pytorch-warmup',
     'sentencepiece',