diff --git a/llmtune/pydantic_models/config_model.py b/llmtune/pydantic_models/config_model.py
index 1095656..e0f9976 100644
--- a/llmtune/pydantic_models/config_model.py
+++ b/llmtune/pydantic_models/config_model.py
@@ -1,7 +1,7 @@
 from typing import List, Literal, Optional, Union
 
 import torch
-from pydantic import BaseModel, ConfigDict, Field, FilePath, validator
+from pydantic import BaseModel, Field, FilePath, validator
 
 
 # TODO: Refactor this into multiple files...