topic_modelling / requirements_gpu.txt
seanpedrickcase's picture
Improved initial clean options. Now has option to return embeddings only.
89c4d20
raw
history blame contribute delete
545 Bytes
gradio==4.44.1
boto3
transformers==4.41.2
accelerate==0.26.1
bertopic==0.16.2
spacy==3.8.0
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==17.0.0
openpyxl==3.1.3
Faker==22.2.0
presidio_analyzer==2.2.354
presidio_anonymizer==2.2.354
scipy==1.11.4
polars==0.20.6
llama-cpp-python==0.2.87 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
torch --index-url https://download.pytorch.org/whl/cu121
sentence-transformers==3.0.1
numpy==1.26.4