Skip to content
Snippets Groups Projects
Commit 61ef4c7c authored by Zhengxiao Du's avatar Zhengxiao Du
Browse files

Add parallel_output argument for GLMModel

parent b4eaba16
No related branches found
No related tags found
No related merge requests found
...@@ -20,8 +20,7 @@ class BlockPositionEmbeddingMixin(BaseMixin): ...@@ -20,8 +20,7 @@ class BlockPositionEmbeddingMixin(BaseMixin):
class GLMModel(BaseModel): class GLMModel(BaseModel):
def __init__(self, args, transformer=None, parallel_output=True): 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', self.add_mixin('block_position_embedding',
BlockPositionEmbeddingMixin(args.max_sequence_length, args.hidden_size) BlockPositionEmbeddingMixin(args.max_sequence_length, args.hidden_size)
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment