From 89ea243c74c33dd0fa072256e58e7c964df4bc72 Mon Sep 17 00:00:00 2001
From: limiteinductive <benjamintrom@gmail.com>
Date: Fri, 15 Dec 2023 18:02:32 +0100
Subject: [PATCH] add tomli to training dependancy

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 99be814eb..17d2e93c5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,6 +26,7 @@ training = [
     # https://github.com/TimDettmers/bitsandbytes/pull/525
     "scipy>=1.11.4",
     "datasets>=2.15.0",
+    "tomli>=2.0.1",
 ]
 test = [
     "diffusers>=0.24.0",