Skip to content

Commit

Permalink
Upperbound tf to 2.16 because of keras 3 and transformers compatibility
Browse files Browse the repository at this point in the history
  • Loading branch information
RobertSamoilescu committed Nov 28, 2024
1 parent 39adc9f commit 5ec4b7b
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,13 +17,15 @@ def readme():
'numba>=0.50.0, !=0.54.0, <0.60.0', # Avoid 0.54 due to: https://github.com/SeldonIO/alibi/issues/466
],

'tensorflow': ['tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.19.0'],
# `keras 3` becomes the default for `tensorflow >= 2.16.0``
# which is not yet supported by `transformers`
'tensorflow': ['tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.16.0'],
'torch': ['torch>=1.9.0, <3.0.0'],
'all': [
'ray>=0.8.7, <3.0.0',
'shap>=0.40.0, <0.44.0',
'numba>=0.50.0, !=0.54.0, <0.60.0',
'tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.19.0',
'tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.16.0',
'torch>=1.9.0, <3.0.0'
]
}
Expand Down

0 comments on commit 5ec4b7b

Please sign in to comment.