huggingface / huggingface/neuralcoref

GPU support - cuda 11.1 - TypeError: Unsupported type <class 'numpy.ndarray'>

Open
#345 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
C
Stars
2.9k
Forks
467
PR merge metrics
No merged PRs in 30d

Description

Example:
``` python
import spacy

spacy.require_gpu()
>> True

nlp = spacy.load("en_core_web_sm")
doc = nlp("this is my example text")
print(doc)
>> this is my example text

import neuralcoref
neuralcoref.add_to_pipe(nlp)
>>

doc = nlp("this is my example text")
>> Traceback (most recent call last):
File "", line 1, in
File "/home/brj/.local/share/virtualenvs/spacy-ozIRu_0L/lib/python3.8/site-packages/spacy/language.py", line 445, in __call__
doc = proc(doc, **component_cfg.get(name, {}))
File "neuralcoref.pyx", line 593, in neuralcoref.neuralcoref.NeuralCoref.__call__
File "neuralcoref.pyx", line 720, in neuralcoref.neuralcoref.NeuralCoref.predict
File "neuralcoref.pyx", line 908, in neuralcoref.neuralcoref.NeuralCoref.get_mention_embeddings
File "neuralcoref.pyx", line 899, in neuralcoref.neuralcoref.NeuralCoref.get_average_embedding
File "cupy/_core/core.pyx", line 1591, in cupy._core.core.ndarray.__array_ufunc__
File "cupy/_core/_kernel.pyx", line 1218, in cupy._core._kernel.ufunc.__call__
File "cupy/_core/_kernel.pyx", line 138, in cupy._core._kernel._preprocess_args
File "cupy/_core/_kernel.pyx", line 124, in cupy._core._kernel._preprocess_arg
TypeError: Unsupported type

# printing versions
import cupy
spacy.__version__
>> 2.3.7
neuralcoref.__version__
>> 4.1.0
cupy.__version__
>> 10.4.0
```

Everything works fine if I run this without `spacy.require_gpu()`.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.