tbitai commited on
Commit
fc8fc8a
1 Parent(s): 834cfb5

Restore TF version, retaining Keras downgrade

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  # TODO: Pin versions
2
  huggingface_hub
3
- tensorflow<2.16 # <2.16 required by sentence-transformers
4
- tf-keras # Required by tensorflow<2.16
5
  numpy
6
  sentence-transformers
 
1
  # TODO: Pin versions
2
  huggingface_hub
3
+ tensorflow
4
+ tf-keras # Required by sentence-transformers. Also set the env var TF_USE_LEGACY_KERAS=1.
5
  numpy
6
  sentence-transformers