AfVaCL / requirements.txt
tugaa's picture
Create requirements.txt
5dd8411 verified
langchain
langchain-openai
langchain-community
langchain-experimental
# TRL and dependencies
trl
transformers
torch # GPU版推奨: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 (CUDA versionに合わせる)
accelerate
peft
bitsandbytes # for 8-bit/4-bit quantization
# Database
psycopg2-binary # または psycopg2
sqlalchemy # ORM (推奨)
# Vector DB extension for SQLAlchemy (optional but helpful)
# sqlalchemy-pgvector
# Message Queue
redis
# Vectorization
sentence-transformers
# Scheduling
apscheduler
# Utilities
python-dotenv
tiktoken
fastapi # for API (optional)
uvicorn # for API (optional)