explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-19 04:32:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use thinc.util.has_torch

Browse Source
This commit is contained in:
Paul O'Leary McCann 2022-05-24 16:02:39 +09:00
parent b1118cee58
commit 5cbc9f4573
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/ml/models/__init__.py
View File

@ -7,10 +7,8 @@ from .textcat import * # noqa
from .tok2vec import * # noqa
# some models require Torch
try:
import torch
from thinc.util import has_torch
if has_torch:
from .coref import * #noqa
from .span_predictor import * #noqa
except ImportError:
pass