File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
examples/models/llama/source_transformation Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -148,7 +148,8 @@ def fuse_ln_linear(
148148 linear .bias .data = linear .bias .data .to (dtype = torch .float32 ) + torch .matmul (
149149 # pyre-fixme[6]: For 2nd argument expected `Tensor` but got
150150 # `Union[Tensor, Module]`.
151- W_ , layernorm .bias .to (dtype = torch .float32 )
151+ W_ ,
152+ layernorm .bias .to (dtype = torch .float32 ),
152153 )
153154 linear .bias .data = linear .bias .data .to (linear_dtype )
154155
Original file line number Diff line number Diff line change @@ -1635,7 +1635,8 @@ def plan(self) -> ExecutionPlan:
16351635 # empty list.
16361636 non_const_buffer_sizes = typing .cast (
16371637 # pyre-fixme[29]: `Union[BoundMethod[typing.Callable(torch._C.TensorB...
1638- List [int ], self .module .meta ["non_const_buffer_sizes" ]
1638+ List [int ],
1639+ self .module .meta ["non_const_buffer_sizes" ],
16391640 ),
16401641 container_meta_type = self .container_meta_type ,
16411642 )
You can’t perform that action at this time.
0 commit comments