black

nyu-mll · Jan 28, 2021 · 58b5b18 · 58b5b18
1 parent 35e127b
commit 58b5b18
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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)