diff --git a/zshot/utils/download_models.py b/zshot/utils/download_models.py index 15ae06b..283140e 100644 --- a/zshot/utils/download_models.py +++ b/zshot/utils/download_models.py @@ -1,5 +1,5 @@ from zshot.evaluation.dataset import load_few_rel_zs, load_medmentions_zs, load_ontonotes_zs -from zshot.linker import LinkerRegen, LinkerSMXM, LinkerTARS +from zshot.linker import LinkerRegen, LinkerSMXM, LinkerTARS, LinkerGLINER from zshot.mentions_extractor import MentionsExtractorFlair from zshot.mentions_extractor.utils import ExtractorType from zshot.relation_extractor.relation_extractor_zsrc import RelationsExtractorZSRC @@ -30,6 +30,10 @@ def load_all(): LinkerTARS().load_models() except RuntimeError: pass + try: + LinkerGLINER().load_models() + except RuntimeError: + pass try: RelationsExtractorZSRC().load_models() except RuntimeError: