Fix cuda version for spaCy

ljvmiranda921 · Jan 5, 2025 · a15d03a · a15d03a
1 parent 2a1b3d9
commit a15d03a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/models/v0.2.0/requirements.txt b/models/v0.2.0/requirements.txt
@@ -1,10 +1,9 @@
 huggingface-hub==0.27.0
-spacy==3.8.3
+spacy[cuda12x]
 spacy-transformers==1.3.5
 torch==2.5.1
 transformers==4.36.2
 spacy-huggingface-hub
-cupy-cuda113
 build
 pip==22.0.2
 sentencepiece