add cobra

2024-05-05 19:56:00 +08:00
parent 7352c07d59
commit b7054cba4d
7 changed files with 163 additions and 70 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,31 @@
+einops==0.6.1
+einops-exts==0.0.4
+gradio_client==0.2.9
+OpenNMT-py==2.2.0
+deepspeed==0.12.3
+peft==0.4.0
+ninja
+torchaudio
+rdkit-pypi>=2021.03.2
+accelerate==0.24.1
+transformers==4.32.1
+bitsandbytes==0.41.0
+scikit-learn==1.2.2
+httpx==0.24.0
+tokenizers>=0.12.1
+markdown2[all]
+wandb
+shortuuid
+tiktoken
+sentencepiece==0.1.99
+torch
+torchvision
+numpy>=1.19.5
+pandas>=1.2.4
+matplotlib>=3.5.3
+opencv-python==4.5.5.64
+huggingface-hub>=0.11.0
+tensorboardX
+SmilesPE==0.0.3
+albumentations @ git+https://github.com/albumentations-team/albumentations@37e714fd2e326f6f88778e425f98c2de8c8d5372
+timm @ git+https://github.com/rwightman/pytorch-image-models.git@54a6cca27a9a3e092a07457f5d56709da56e3cf5