File size: 764 Bytes
8d4809b
8b9d915
cba9d47
 
8b9d915
45adc0d
8d4809b
8b9d915
 
 
8d4809b
 
 
 
 
 
 
45adc0d
32eed14
 
8b9d915
45adc0d
 
8b9d915
 
32eed14
 
45adc0d
 
 
 
32eed14
 
45adc0d
8b9d915
cba9d47
8d4809b
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# Core Libraries
torch  # Latest CUDA-enabled version
torchvision
torchaudio
transformers>=4.36.0
sentence-transformers
faiss-gpu==1.7.2  # Use GPU-optimized FAISS
accelerate>=0.26.0
bitsandbytes>=0.41.0
numpy<2.0  # Downgrade for compatibility with FAISS

# Hugging Face and LangChain
huggingface_hub>=0.12.0  # Specify version for compatibility
langchain>=0.2.0  # Ensure latest compatible version
langchain-community>=0.2.0  # Align with updated imports

# Gradio for Interface
gradio

# Machine Learning dependencies
tf-keras
tensorflow
scikit-learn
optimum
auto-gptq

# Tracking and visualization
wandb
matplotlib
seaborn
pandas

# Development dependencies
pytest
sentencepiece
protobuf
ipython  # Required for IPython display

# Utilities
python-dotenv
tqdm