PULSE-7B / requirements.txt
paralym's picture
Update requirements.txt
7df6445 verified
raw
history blame
441 Bytes
ninja
opencv-python
open_clip_torch
fastapi
gradio==4.35.0
gradio_client==1.0.1
markdown2[all]
numpy==1.26.4
requests
sentencepiece
torch==2.1.2
torchvision==0.16.2
uvicorn
wandb==0.16.5
deepspeed==0.12.2
peft==0.4.0
accelerate>=0.29.1
tokenizers~=0.15.2
transformers
bitsandbytes==0.41.0
scikit-learn==1.2.2
sentencepiece~=0.1.99
einops==0.6.1
einops-exts==0.0.4
pydantic>=2.0
timm
hf_transfer
decord
datasets
tyro
scipy
rouge
urllib3~=2.0