# 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