final-thesis / environment.yml
deddoggo's picture
install dependencies
af7c566
raw
history blame contribute delete
981 Bytes
name: llm-rag-env # Đặt tên cho môi trường của bạn
channels:
- nvidia # Kênh cho cudatoolkit (nếu Space của bạn có GPU)
- conda-forge # Kênh cho faiss và các gói khác
- defaults
dependencies:
- python=3.11 # Hoặc phiên bản Python bạn muốn
- pip
# Quan trọng: Đảm bảo phiên bản cudatoolkit phù hợp với GPU trên Space
- cudatoolkit=11.8
# FAISS từ conda-forge
- conda-forge::faiss=1.9.0=*cuda* # Dấu *cuda* để Conda tìm bản build GPU
# Các gói khác
- huggingface_hub==0.31.2 # Hoặc phiên bản mới nhất bạn muốn
- gradio==5.31.0 # Hoặc phiên bản mới nhất bạn muốn
- sentence-transformers==4.1.0 # Thêm ==phiên_bản
- unsloth==2025.5.7 # Thêm ==phiên_bản
- transformers # Thêm ==phiên_bản
- rank_bm25==0.2.2
- numpy==1.26.4
# Nếu có gói nào chỉ có trên pip và không có trên Conda channels
# - pip:
# - ten-goi-pip