Update requirements.txt

requirements.txt

@@ -1,61 +1,60 @@
-absl-py
-wheel
-accelerate==0.30.1
-annotated-types==0.7.0
-bitsandbytes==0.42.0
-certifi==2024.2.2
-charset-normalizer==3.3.2
-deepspeed==0.14.2
-einops==0.8.0
-filelock==3.14.0
-flash-attn==2.6.1
-fsspec==2024.5.0
-grpcio
-hjson==3.1.0
-huggingface-hub==0.23.1
-idna==3.7
-importlib_metadata==7.1.0
-Jinja2==3.1.4
-joblib==1.4.2
-MarkupSafe==2.1.5
-mpmath==1.3.0
-networkx==3.2.1
-ninja==1.11.1.1
-numpy==1.26.4
-packaging==24.0
-pandas==2.2.2
-peft==0.11.1
-pillow==10.3.0
-protobuf==3.20.3
-psutil==5.9.8
-py-cpuinfo==9.0.0
-pydantic==2.7.1
-pydantic_core==2.18.2
-pynvml==11.5.0
-python-dateutil==2.9.0.post0
-pytz==2024.1
-PyYAML==6.0.1
-regex==2024.5.15
-requests==2.32.2
-safetensors==0.4.3
-scikit-learn==1.5.1
-scipy==1.13.1
-six
-sympy==1.12
-tensorboard==2.17.0
-tensorboard-data-server==0.7.2
-threadpoolctl==3.5.0
-tiktoken==0.7.0
-tokenizers==0.19.1
-torch==2.3.0
-torchaudio==2.3.0
-torchvision==0.18.0
-tqdm==4.66.4
-transformers==4.41.1
-transformers-stream-generator==0.0.4
-triton==2.3.0
-typing_extensions==4.12.0
-tzdata==2024.1
-urllib3==2.2.1
-Werkzeug
+absl-py
+wheel
+accelerate==0.30.1
+annotated-types==0.7.0
+bitsandbytes==0.42.0
+certifi==2024.2.2
+charset-normalizer==3.3.2
+deepspeed==0.14.2
+einops==0.8.0
+filelock==3.14.0
+fsspec==2024.5.0
+grpcio
+hjson==3.1.0
+huggingface-hub==0.23.1
+idna==3.7
+importlib_metadata==7.1.0
+Jinja2==3.1.4
+joblib==1.4.2
+MarkupSafe==2.1.5
+mpmath==1.3.0
+networkx==3.2.1
+ninja==1.11.1.1
+numpy==1.26.4
+packaging==24.0
+pandas==2.2.2
+peft==0.11.1
+pillow==10.3.0
+protobuf==3.20.3
+psutil==5.9.8
+py-cpuinfo==9.0.0
+pydantic==2.7.1
+pydantic_core==2.18.2
+pynvml==11.5.0
+python-dateutil==2.9.0.post0
+pytz==2024.1
+PyYAML==6.0.1
+regex==2024.5.15
+requests==2.32.2
+safetensors==0.4.3
+scikit-learn==1.5.1
+scipy==1.13.1
+six
+sympy==1.12
+tensorboard==2.17.0
+tensorboard-data-server==0.7.2
+threadpoolctl==3.5.0
+tiktoken==0.7.0
+tokenizers==0.19.1
+torch==2.3.0
+torchaudio==2.3.0
+torchvision==0.18.0
+tqdm==4.66.4
+transformers==4.41.1
+transformers-stream-generator==0.0.4
+triton==2.3.0
+typing_extensions==4.12.0
+tzdata==2024.1
+urllib3==2.2.1
+Werkzeug
 zipp==3.18.2