Demo_ProTrek_650M_UniRef50 / requirements.txt
LTEnjoy's picture
Update requirements.txt
7994a0a verified
torch==2.0.1
gradio==4.37.1
torchmetrics==0.9.3
pytorch-lightning==2.1.3
transformers==4.28.0
scikit-learn==1.4.0
tabulate==0.9.0