requirements.txt

--extra-index-url https://download.pytorch.org/whl/cpu
diffusers==0.35.2
https://github.com/huggingface/optimum-intel/archive/2c48d6430c265ac259c1b264f3e2c4025cdd7b76.tar.gz#egg=optimum-intel
numpy==1.26.4; platform_system == "Darwin" and platform_machine == "x86_64"
safetensors==0.6.2; platform_system == "Darwin" and platform_machine == "x86_64"
pytest==9.0.2
transformers==4.55.4
hf_transfer==0.1.9
gguf==0.17.1

# rag requirements
langchain_community==0.4.1
langchain-core==1.2.11

# requirements for specific models
# - Qwen/Qwen-7B
# - Qwen/Qwen-7B-Chat
einops==0.8.2
# - openbmb/MiniCPM-V-2
torchvision==0.17.2; platform_system == "Darwin" and platform_machine == "x86_64"
torchvision==0.23.0; platform_system != "Darwin" or platform_machine != "x86_64"
# - openbmb/MiniCPM-V-2
timm==1.0.25
# - openai/whisper-base
librosa==0.11.0
soundfile==0.13.1
datasets==4.1.1; sys_platform == "linux"
datasets==3.6.0; sys_platform != "linux"
sentence-transformers>=2.2.2,<=5.2.2
torchcodec==0.7.0; sys_platform == "linux"
rouge==1.0.1
# - microsoft/Phi-4-multimodal-instruct
peft==0.18.1
# - openbmb/MiniCPM-o-2_6
vocos==0.1.0
torchaudio==2.8.0
vector-quantize-pytorch==1.27.20