From b6fecae91a22b7b969a3991f56b6771d36082486 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Wed, 18 Oct 2023 21:07:47 -0700
Subject: [PATCH] fix another pydantic 2 migration error

---
 train_decoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_decoder.py b/train_decoder.py
index 53c250f..2218655 100644
--- a/train_decoder.py
+++ b/train_decoder.py
@@ -636,7 +636,7 @@ def initialize_training(config: TrainDecoderConfig, config_path):
         inference_device=accelerator.device,
         evaluate_config=config.evaluate,
         condition_on_text_encodings=conditioning_on_text,
-        **config.train.dict(),
+        **config.train.model_dump(),
     )
     
 # Create a simple click command line interface to load the config and start the training