From 61ef4c7c5a19ee909b94c565d1288eccc8b1994b Mon Sep 17 00:00:00 2001
From: Zhengxiao Du <zx-du20@mails.tsinghua.edu.cn>
Date: Mon, 8 Nov 2021 11:16:18 +0800
Subject: [PATCH] Add parallel_output argument for GLMModel
---
SwissArmyTransformer/model/glm_model.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/SwissArmyTransformer/model/glm_model.py b/SwissArmyTransformer/model/glm_model.py
index 01aa075..4ad68bf 100644
--- a/SwissArmyTransformer/model/glm_model.py
+++ b/SwissArmyTransformer/model/glm_model.py
@@ -20,8 +20,7 @@ class BlockPositionEmbeddingMixin(BaseMixin):
class GLMModel(BaseModel):
def __init__(self, args, transformer=None, parallel_output=True):
- super().__init__(args, transformer=transformer, parallel_output=parallel_output
- )
+ super().__init__(args, transformer=transformer, parallel_output=parallel_output)
self.add_mixin('block_position_embedding',
BlockPositionEmbeddingMixin(args.max_sequence_length, args.hidden_size)
)
--
GitLab