mirror of https://github.com/explosion/spaCy.git
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
import warnings
|
|
warnings.filterwarnings("ignore", message="numpy.dtype size changed")
|
|
warnings.filterwarnings("ignore", message="numpy.ufunc size changed")
|
|
|
|
# These are imported as part of the API
|
|
from thinc.neural.util import prefer_gpu
|
|
|
|
from .cli.info import info as cli_info
|
|
from .glossary import explain
|
|
from .about import __version__
|
|
from .errors import Warnings, deprecation_warning
|
|
from . import util
|
|
|
|
|
|
def load(name, **overrides):
|
|
depr_path = overrides.get('path')
|
|
if depr_path not in (True, False, None):
|
|
deprecation_warning(Warnings.W001.format(path=depr_path))
|
|
return util.load_model(name, **overrides)
|
|
|
|
|
|
def blank(name, **kwargs):
|
|
LangClass = util.get_lang_class(name)
|
|
return LangClass(**kwargs)
|
|
|
|
|
|
def info(model=None, markdown=False, silent=False):
|
|
return cli_info(model, markdown, silent)
|
|
|
|
def require_gpu():
|
|
from thinc.v2v import Model
|
|
from thinc.neural.ops import CupyOps
|
|
if CupyOps.xp is None:
|
|
raise ValueError(
|
|
"GPU is not accessible. Check your LD_LIBRARY_PATH enironment variable "
|
|
"and check that thinc was installed with GPU, e.g. thinc[cuda]")
|
|
Model.Ops = CupyOps
|
|
Model.ops = CupyOps()
|
|
return True
|