HakimAiV2 / requirements.txt
scdrand23's picture
major change, used biomed_llama 7b
3bf392c
# --no-cache-dir
pillow==9.4.0
opencv-python==4.8.1.78
# torch==2.5.1
pyyaml==6.0.1
json_tricks==3.17.3
yacs==0.1.8
scikit-learn==1.3.1
pandas==2.0.3
timm==0.4.12
numpy==1.26.4
einops==0.8.0
fvcore==0.1.5.post20221221
# transformers==4.34.0
ftfy==6.1.1
regex==2023.10.3
nltk==3.8.1
pydicom
nibabel
SimpleITK
vision-datasets==0.2.2
cython==3.0.2
pycocotools==2.0.7
diffdist==0.1
scikit-image==0.21.0
mup==1.0.0
# accelerate
kornia==0.7.0
infinibatch==0.1.1
open-clip-torch==2.26.1
git+https://github.com/MaureenZOU/detectron2-xyz.git
antlr4-python3-runtime==4.9.3
appdirs==1.4.4
black==21.4b2
cloudpickle==3.0.0
hjson==3.1.0
huggingface-hub>=0.19.3,<1.0
hydra-core==1.3.2
imageio==2.35.1
iopath==0.1.9
mypy-extensions==1.0.0
ninja==1.11.1.1
omegaconf==2.3.0
pathspec==0.12.1
portalocker==2.10.1
py-cpuinfo==9.0.0
pydantic==1.10.18
pydot==3.0.1
tabulate==0.9.0
termcolor==2.4.0
transformers>=4.36.0
accelerate>=0.25.0
bitsandbytes>=0.41.1
sentencepiece>=0.1.99
huggingface-hub>=0.19.3,<1.0
tokenizers>=0.14.1
#
# torch>=2.0.0
# transformers>=4.34.0
# gradio>=4.40.0
# Pillow>=9.0.0
# numpy>=1.21.0
tqdm>=4.65.0
#
safetensors>=0.4.0
# accelerate>=0.25.0