team-qbits / requirements.txt
ovinduG's picture
Upload folder using huggingface_hub
e55c8a4 verified
Raw History Blame Contribute Delete
530 Bytes
unsloth
# Sinhala MMLU Shared Task - Team QBitsLK
# Verified install: pip install -r requirements.txt && python inference.py
#
# Inference needs ONE GPU with >= 20 GB of memory (the model is 16 GB in bf16).
# Reference machine: 1x NVIDIA A40 48 GB, driver 595.91, CUDA 13.2, Python 3.12.
torch>=2.6
transformers==5.5.0
tokenizers>=0.22.0,<=0.23.0
huggingface_hub>=1.5.0,<2.0
accelerate>=1.0
timm # Gemma 4 ships a multimodal processor; loading it requires timm
numpy
hf_transfer # faster weight download; optional