diff --git a/SwissArmyTransformer/model/glm_model.py b/SwissArmyTransformer/model/glm_model.py index 01aa07554d69cc9caee0479f99efbe5098aa8f3c..4ad68bf243bed66f7005dc1cf09d4a858450e851 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) )