NameError: name 'pydantic' is not defined

пытаюсь с помощью python работать с replicate.com, вроде все сделал как в документации, но при попытке запустить генерацию выдает ошибку - NameError: name 'pydantic' is not defined, понятия не имею зачем это вообще и как связанно с тем что я делаю. Помогите пожалуйста, нигде не удалось найти пока решение..

from dotenv import load_dotenv
import replicate
load_dotenv()


output = replicate.run(
  "stability-ai/stable-diffusion:27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478",
  input={
    "width": 512,
    "height": 512,
    "prompt": 'cosmos air'
  }
)
print(output)

в итоге:

Traceback (most recent call last):
  File "C:\Games\PictureGenBot\test.py", line 2, in <module>
    import replicate
  File "C:\Games\PictureGenBot\venv\Lib\site-packages\replicate\__init__.py", line 1, in <module>
    from replicate.client import Client
  File "C:\Games\PictureGenBot\venv\Lib\site-packages\replicate\client.py", line 20, in <module>
    from replicate.collection import Collections
  File "C:\Games\PictureGenBot\venv\Lib\site-packages\replicate\collection.py", line 5, in <module>
    from replicate.model import Model
  File "C:\Games\PictureGenBot\venv\Lib\site-packages\replicate\model.py", line 21, in <module>
    class Model(Resource):
  File "C:\Games\PictureGenBot\venv\Lib\site-packages\replicate\model.py", line 26, in Model
    _client: "Client" = pydantic.PrivateAttr()
                        ^^^^^^^^
NameError: name 'pydantic' is not defined