Browse Source

[MNT] update dependency

tags/v0.3.2
bxdd 2 years ago
parent
commit
29d372da5d
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      setup.py

+ 4
- 4
setup.py View File

@@ -108,12 +108,12 @@ if __name__ == "__main__":
], ],
"full": [ "full": [
# The default full requirements for learnware package # The default full requirements for learnware package
"torch==2.1.0",
"torchvision==0.16.0",
"torch==2.0.1",
"torchvision==0.15.2",
"torch-optimizer>=0.3.0", "torch-optimizer>=0.3.0",
"lightgbm>=3.3.0", "lightgbm>=3.3.0",
"sentence_transformers>=2.2.2",
"fast_pytorch_kmeans>=0.2.0.1",
"sentence_transformers==2.2.2",
"fast_pytorch_kmeans==0.2.0.1",
], ],
}, },
classifiers=[ classifiers=[


Loading…
Cancel
Save