MoraxCheng's picture
Add error handling for Zero GPU initialization and fallback to CPU mode; update Gradio version in requirements
86b2ecb
raw
history blame contribute delete
372 Bytes
torch>=1.12.0
transformers==4.17.0
tokenizers==0.11.6
spaces>=0.19.0
numpy>=1.21.0,<1.24.0
pandas>=1.3.0,<2.0.0
scipy>=1.7.0,<1.11.0
scikit-learn>=1.0.0,<1.2.0
tqdm
biopython>=1.78,<2.0.0
matplotlib
seaborn
gradio>=4.0.0,<5.0.0
datasets==1.18.3
wandb==0.12.11
jax>=0.3.25
jaxlib>=0.4.6
flax>=0.6.0
optax
sentencepiece==0.1.96
aiohttp>=3.8.0,<3.9.0
sentry-sdk>=1.3.1,<2.0.0