add bitsandbytes for minicpm v2

BBC-Esq · Aug 7, 2024 · 258d0b8 · 258d0b8
1 parent d02a06e
commit 258d0b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/setup_windows.py b/src/setup_windows.py
@@ -144,7 +144,6 @@ def install_libraries(libraries):
 priority_libraries = [
     "flash_attn @ https://github.com/bdashore3/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.2.2cxx11abiFALSE-cp311-cp311-win_amd64.whl",
     "torch==2.2.2",
-    "triton @ https://github.com/jakaline-dev/Triton_win/releases/download/3.0.0/triton-3.0.0-cp311-cp311-win_amd64.whl#sha256=2c78f5f85cf88d46eb9664c23691052d6c153a6043656fc15c50a0d13bc5565c",
     "whisper_s2t @ git+https://github.com/shashikg/WhisperS2T.git@e7f7e6dbfdc7f3a39454feb9dd262fd3653add8c",
     "WhisperSpeech @ git+https://github.com/BBC-Esq/WhisperSpeech.git@41c9accb7d9ac1e4e5f5c110a4a973c566c56fd8"
 ]
@@ -161,6 +160,7 @@ def install_libraries(libraries):
     "av==12.3.0",
     "backoff==2.2.1",
     "beautifulsoup4==4.12.3",
+    "bitsandbytes==0.43.1",
     "certifi==2024.7.4",
     "chardet==5.2.0",
     "charset-normalizer==3.3.2",