add tomli to training dependancy

finegrain-ai · Dec 20, 2023 · 0f56043 · 0f56043
1 parent 22ce3fd
commit 0f56043
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",