Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

pyproject.toml: add a note about scipy + bitsandbytes

Browse source
This commit is contained in:
Cédric Deltheil 2023-10-11 15:39:36 +02:00 committed by Cédric Deltheil
parent ac631bfb2e
commit e35dce825f
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -23,6 +23,8 @@ loguru = {version = "^0.7.0", optional = true}
bitsandbytes = {version = "^0.41.0", optional = true} bitsandbytes = {version = "^0.41.0", optional = true}
prodigyopt = {version = "^1.0", optional = true} prodigyopt = {version = "^1.0", optional = true}
pydantic = {version = "~2.0.3", optional = true} pydantic = {version = "~2.0.3", optional = true}
# Added scipy as a work around until this PR gets merged:
# https://github.com/TimDettmers/bitsandbytes/pull/525
scipy = {version = "^1.11.1", optional = true} scipy = {version = "^1.11.1", optional = true}
torchvision = {version = "^0.16.0", optional = true} torchvision = {version = "^0.16.0", optional = true}
diffusers = {version = "^0.21.0", optional = true} diffusers = {version = "^0.21.0", optional = true}