diff --git a/README.md b/README.md index 4b147b3d..ed52a337 100644 --- a/README.md +++ b/README.md @@ -89,12 +89,14 @@ You can install NeMo-Curator from PyPi, from source or get it through the NeMo F To install the CPU-only modules: ```bash +pip install cython pip install nemo-curator ``` To install the CPU and CUDA-accelerated modules: ```bash +pip install cython pip install --extra-index-url https://pypi.nvidia.com nemo-curator[cuda12x] ``` @@ -112,12 +114,14 @@ pip install --extra-index-url https://pypi.nvidia.com nemo-curator[cuda12x] To install the CPU-only modules: ```bash + pip install cython pip install . ``` To install the CPU and CUDA-accelerated modules: ```bash + pip install cython pip install --extra-index-url https://pypi.nvidia.com ".[cuda12x]" ``` diff --git a/setup.py b/setup.py index 6035ebdc..ca12e147 100644 --- a/setup.py +++ b/setup.py @@ -65,7 +65,6 @@ "presidio-anonymizer==2.2.351", "usaddress==0.5.10", "nemo_toolkit[nlp]>=1.23.0", - "Cython", "crossfit @ git+https://github.com/rapidsai/crossfit.git@0cc2993", # Numpy 2.0 breaks with spacy https://github.com/explosion/spaCy/issues/13528 # TODO: Remove when issue is fixed