Commit 7b8abfa1 authored by Wael Ramadan's avatar Wael Ramadan
Browse files

use_gpu false by default

parent 73640c72
Pipeline #6893 passed with stage
in 5 minutes and 32 seconds
......@@ -25,7 +25,8 @@ from texta_mlp.settings import (
SUPPORTED_ANALYZERS,
DEFAULT_ANALYZERS,
ENTITY_MAPPER_DATA_URLS,
CONCATENATOR_DATA_FILES
CONCATENATOR_DATA_FILES,
USE_GPU
)
......@@ -39,7 +40,7 @@ class MLP:
resource_dir: str = DEFAULT_RESOURCE_DIR,
ner_model_langs: list = CUSTOM_NER_MODEL_LANGS,
logging_level: str ="error",
use_gpu: bool = True,
use_gpu: bool = USE_GPU,
gpu_device_id: int = 0,
refresh_data: bool = REFRESH_DATA
):
......
......@@ -66,3 +66,6 @@ STANZA_NER_SUPPORT = ("ar", "zh", "nl", "en", "fr", "de", "ru", "es", "uk")
# Here we add langs that will have custom ner models.
CUSTOM_NER_MODEL_LANGS = ["et"]
# Use gpu for pytorch
USE_GPU = parse_bool_env("TEXTA_MLP_USE_GPU", False)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment