Skip to content
Snippets Groups Projects
Commit 1c7de299 authored by Michał Marcińczuk's avatar Michał Marcińczuk
Browse files

Poleval model with default merged sequence generator. Version bump to 0.8.0.

parent 95276d66
No related branches found
No related tags found
1 merge request!41Dev v07
Pipeline #9146 canceled
...@@ -24,7 +24,6 @@ def main(model_path: str, input: str, output: str, device: str, do_merge_sentenc ...@@ -24,7 +24,6 @@ def main(model_path: str, input: str, output: str, device: str, do_merge_sentenc
print("Loading the NER model ...") print("Loading the NER model ...")
t0 = time.perf_counter() t0 = time.perf_counter()
ner = PolDeepNer2(model_path, device=device, tokenizer=TokenizerFast()) ner = PolDeepNer2(model_path, device=device, tokenizer=TokenizerFast())
ner.model.config.sequence_generator = "context-window"
time_load = time.perf_counter() - t0 time_load = time.perf_counter() - t0
time_preprocess, time_ner, data_size = 0, 0, 0 time_preprocess, time_ner, data_size = 0, 0, 0
......
...@@ -17,7 +17,7 @@ install_requires = [ ...@@ -17,7 +17,7 @@ install_requires = [
setuptools.setup( setuptools.setup(
name="poldeepner2", name="poldeepner2",
version="0.8.1-wip", version="0.8.0",
author="Michał Marcińczuk", author="Michał Marcińczuk",
author_email="marcinczuk@gmail.com", author_email="marcinczuk@gmail.com",
description="PolDeepNer2 is a tool for sequence labeling tasks based on " description="PolDeepNer2 is a tool for sequence labeling tasks based on "
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment