Magma-UI / requirements.txt
jw2yang's picture
update
bf1344c
raw
history blame contribute delete
628 Bytes
torch==2.4.0
torchvision
pytorch-lightning>=1.0.8
transformers @ git+https://github.com/jwyang/transformers.git@dev/jwyang-v4.44.1
tokenizers>=0.15.0
sentencepiece==0.1.99
shortuuid
accelerate==0.34.2
peft==0.4.0
bitsandbytes==0.44.1
pydantic>=2.0
markdown2[all]
numpy
scikit-learn==1.5.0
gradio==4.44.1
gradio_client
spaces
requests
httpx
uvicorn
fastapi
einops==0.6.1
einops-exts==0.0.4
timm==0.9.12
tensorflow==2.15.0
tensorflow_datasets==4.9.3
tensorflow_graphics==2021.12.3
draccus
pyav
numba
loguru
sacrebleu
evaluate
sqlitedict
open_clip_torch
easyocr
paddleocr
paddlepaddle==2.6.2
supervision==0.18.0
ultralytics==8.3.78