Fix vocab size settings
@@ -66,6 +66,7 @@ class PyTT_TokenVectorEncoder(Pipe):
config_cls = get_pytt_config(name)
model_cls = get_pytt_model(name)
# Need to match the name their constructor expects.
+ if "vocab_size" in cfg["pytt_config"]:
vocab_size = cfg["pytt_config"]["vocab_size"]
cfg["pytt_config"]["vocab_size_or_config_json_file"] = vocab_size
pytt_wrap = PyTT_Wrapper(
