Magma-Gaming / requirements.txt
jw2yang's picture
add magma arena
4f00e93
raw
history blame contribute delete
615 Bytes
torch==2.3.1
torchvision==0.18.1
pytorch-lightning>=1.0.8
transformers @ git+https://github.com/jwyang/transformers.git@dev/jwyang-v4.48.2
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_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
supervision==0.18.0
ultralytics==8.3.78
pygame
pyautogui
qwen-vl-utils