topic_modelling / requirements_gpu.txt
seanpedrickcase's picture
Updated package versions in requirements files
5814ab0
raw
history blame contribute delete
613 Bytes
pandas==2.3.3
plotly==6.3.1
scikit-learn==1.7.2
umap-learn==0.5.9.post2
gradio==5.49.1
boto3==1.40.55
transformers==4.57.1
accelerate==1.11.0
torch==2.6.0 --extra-index-url https://download.pytorch.org/whl/cu124
bertopic==0.17.3
spacy==3.8.7
en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0.tar.gz
pyarrow
openpyxl
Faker
presidio_analyzer==2.2.355
presidio_anonymizer==2.2.355
scipy
polars
llama-cpp-python==0.3.4 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
sentence-transformers==5.1.1
spaces==0.42.1
numpy==2.2.6