add tomli to training dependancy

This commit is contained in:
limiteinductive 2023-12-15 18:02:32 +01:00 committed by Benjamin Trom
parent 22ce3fd033
commit 0f560437bc

View file

@ -26,6 +26,7 @@ training = [
# https://github.com/TimDettmers/bitsandbytes/pull/525 # https://github.com/TimDettmers/bitsandbytes/pull/525
"scipy>=1.11.4", "scipy>=1.11.4",
"datasets>=2.15.0", "datasets>=2.15.0",
"tomli>=2.0.1",
] ]
test = [ test = [
"diffusers>=0.24.0", "diffusers>=0.24.0",