PearlIsa commited on
Commit
32eed14
·
verified ·
1 Parent(s): ec50d5a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +23 -22
requirements.txt CHANGED
@@ -1,23 +1,24 @@
 
1
  huggingface_hub==0.25.2
2
- torch
3
- torchvision
4
- torchaudio
5
- transformers
6
- sentence-transformers
7
- faiss-cpu
8
- datasets
9
- peft
10
- pytest
11
- numpy
12
- gradio
13
- wandb
14
- gradio_client
15
- tqdm
16
- python-dotenv
17
- tensorflow
18
- keras
19
- scikit-learn
20
- matplotlib
21
- seaborn
22
- pandas
23
- requests
 
1
+ # Core dependencies
2
  huggingface_hub==0.25.2
3
+ torch==2.1.0
4
+ transformers==4.36.0
5
+ sentence-transformers==2.2.2
6
+ faiss-cpu==1.7.4
7
+ numpy==1.24.3
8
+ gradio==4.19.2
9
+ tqdm==4.66.1
10
+ python-dotenv==1.0.0
11
+
12
+ # Machine Learning dependencies
13
+ tf-keras==2.15.0 # backwards-compatible keras
14
+ tensorflow==2.15.0 # specific version
15
+ scikit-learn==1.3.2
16
+
17
+ # Tracking and visualization
18
+ wandb==0.16.0
19
+ matplotlib==3.8.2
20
+ seaborn==0.13.0
21
+ pandas==2.1.3
22
+
23
+ # Development dependencies
24
+ pytest==7.4.3