From 06c4376bc8d01a42bc9050bc55f2491ee01fdfe2 Mon Sep 17 00:00:00 2001
From: Maja Jablonska <majajjablonska@gmail.com>
Date: Wed, 15 Nov 2023 19:54:58 +1100
Subject: [PATCH] Add missing import in combo_model.py
---
combo/combo_model.py | 2 +-
combo/main.py | 6 +++++-
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/combo/combo_model.py b/combo/combo_model.py
index 79c9f36..d231231 100644
--- a/combo/combo_model.py
+++ b/combo/combo_model.py
@@ -23,7 +23,7 @@ from combo.nn import utils
from combo.nn.utils import get_text_field_mask
from combo.predictors import Predictor
from combo.utils import metrics
-from utils import ConfigurationError
+from combo.utils import ConfigurationError
@Registry.register("semantic_multitask")
diff --git a/combo/main.py b/combo/main.py
index 44a6a30..055ebeb 100755
--- a/combo/main.py
+++ b/combo/main.py
@@ -352,7 +352,11 @@ def _get_ext_vars(finetuning: bool = False) -> Dict:
}
}
},
- "pretrained_tokens": FLAGS.pretrained_tokens,
+ "vocabulary": {
+ "parameters": {
+ "pretrained_files": FLAGS.pretrained_tokens
+ }
+ },
"word_batch_size": int(FLAGS.word_batch_size),
}
--
GitLab