From 58b5b1813a85ee94cc140e9ab1d0535c8b2d27c0 Mon Sep 17 00:00:00 2001 From: Jesse Swanson Date: Thu, 28 Jan 2021 15:53:05 -0800 Subject: [PATCH] black --- jiant/proj/main/modeling/model_setup.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/jiant/proj/main/modeling/model_setup.py b/jiant/proj/main/modeling/model_setup.py index 4e14e0878..f176e56cf 100644 --- a/jiant/proj/main/modeling/model_setup.py +++ b/jiant/proj/main/modeling/model_setup.py @@ -288,9 +288,7 @@ def create_taskmodel(task, jiant_transformers_model, **taskmodel_kwargs) -> Task head_kwargs["vocab_size"] = jiant_transformers_model.config.vocab_size head_kwargs["layer_norm_eps"] = (jiant_transformers_model.config.layer_norm_eps,) head_kwargs["hidden_act"] = jiant_transformers_model.config.hidden_act - head_kwargs["model_arch"] = ModelArchitectures( - jiant_transformers_model.config.model_type - ) + head_kwargs["model_arch"] = ModelArchitectures(jiant_transformers_model.config.model_type) head = JiantHeadFactory()(task, **head_kwargs)