requirements_gpu.txt

gradio # Not specified version due to interaction with spacy - reinstall latest version after requirements.txt load
transformers==4.41.2
accelerate==0.26.1
bertopic==0.16.2
spacy==3.7.4
en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.1/en_core_web_sm-3.7.1.tar.gz
pyarrow==14.0.2
openpyxl==3.1.2
Faker==22.2.0
presidio_analyzer==2.2.354
presidio_anonymizer==2.2.354
scipy==1.11.4
polars==0.20.6
torch --index-url https://download.pytorch.org/whl/cu121
llama-cpp-python==0.2.77 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
numpy==1.26.4
sentence-transformers==3.0.1