Files
chemstructlm/requirements.txt
2024-05-05 19:56:00 +08:00

31 lines
679 B
Plaintext

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