Ver código fonte

Merge pull request #13187 from Mister-Hope/patch-1

feat: update default CUDA version to 12.8, close #13186
Tim Jaeryang Baek 5 meses atrás
pai
commit
2f5abf7374
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      Dockerfile

+ 1 - 1
Dockerfile

@@ -4,7 +4,7 @@
 ARG USE_CUDA=false
 ARG USE_OLLAMA=false
 # Tested with cu117 for CUDA 11 and cu121 for CUDA 12 (default)
-ARG USE_CUDA_VER=cu121
+ARG USE_CUDA_VER=cu128
 # any sentence transformer model; models to use can be found at https://huggingface.co/models?library=sentence-transformers
 # Leaderboard: https://huggingface.co/spaces/mteb/leaderboard 
 # for better performance and multilangauge support use "intfloat/multilingual-e5-large" (~2.5GB) or "intfloat/multilingual-e5-base" (~1.5GB)