Update Requirements.txt
Browse files- Requirements.txt +6 -11
Requirements.txt
CHANGED
|
@@ -4,12 +4,12 @@ torchvision==0.16.0
|
|
| 4 |
torchaudio==2.1.0
|
| 5 |
numpy==1.24.3
|
| 6 |
|
| 7 |
-
# Hugging Face Ecosystem
|
| 8 |
-
transformers
|
| 9 |
datasets==2.14.0
|
| 10 |
-
huggingface_hub
|
| 11 |
-
accelerate
|
| 12 |
-
tokenizers
|
| 13 |
|
| 14 |
# UI Framework
|
| 15 |
gradio==4.8.0
|
|
@@ -36,7 +36,7 @@ scipy==1.11.4
|
|
| 36 |
# Optimization Libraries
|
| 37 |
bitsandbytes==0.41.3
|
| 38 |
einops==0.7.0
|
| 39 |
-
flash-attn==2.3.6
|
| 40 |
|
| 41 |
# Networking & File Handling
|
| 42 |
requests==2.31.0
|
|
@@ -60,11 +60,6 @@ pyyaml==6.0.1
|
|
| 60 |
toml==0.10.2
|
| 61 |
jsonlines==4.0.0
|
| 62 |
|
| 63 |
-
# Development & Testing
|
| 64 |
-
pytest==7.4.3
|
| 65 |
-
black==23.11.0
|
| 66 |
-
flake8==6.1.0
|
| 67 |
-
|
| 68 |
# Memory & Performance
|
| 69 |
psutil==5.9.6
|
| 70 |
memory-profiler==0.61.0
|
|
|
|
| 4 |
torchaudio==2.1.0
|
| 5 |
numpy==1.24.3
|
| 6 |
|
| 7 |
+
# Hugging Face Ecosystem - VERSIONS MISES À JOUR
|
| 8 |
+
transformers>=4.45.0
|
| 9 |
datasets==2.14.0
|
| 10 |
+
huggingface_hub>=0.20.0
|
| 11 |
+
accelerate>=0.25.0
|
| 12 |
+
tokenizers>=0.15.0
|
| 13 |
|
| 14 |
# UI Framework
|
| 15 |
gradio==4.8.0
|
|
|
|
| 36 |
# Optimization Libraries
|
| 37 |
bitsandbytes==0.41.3
|
| 38 |
einops==0.7.0
|
| 39 |
+
# flash-attn==2.3.6 # Commenté car peut causer des problèmes sur certains systèmes
|
| 40 |
|
| 41 |
# Networking & File Handling
|
| 42 |
requests==2.31.0
|
|
|
|
| 60 |
toml==0.10.2
|
| 61 |
jsonlines==4.0.0
|
| 62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
# Memory & Performance
|
| 64 |
psutil==5.9.6
|
| 65 |
memory-profiler==0.61.0
|