| @@ -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=[ | ||||