bitnet-space / requirements.txt
aaurelions's picture
Create requirements.txt
e84e757 verified
raw
history blame contribute delete
424 Bytes
transformers>=4.38.0 # Use a recent version
torch>=2.1.0 # Check compatibility with your hardware/CUDA
gradio>=4.0.0 # Use a recent version of Gradio
accelerate # Often needed for device_map='auto'
bitsandbytes # Might be needed depending on how BitNet is loaded or if using further quantization
# sentencepiece # Often required by tokenizers
# protobuf # Often required by tokenizers