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(
148
148
linear .bias .data = linear .bias .data .to (dtype = torch .float32 ) + torch .matmul (
149
149
# pyre-fixme[6]: For 2nd argument expected `Tensor` but got
150
150
# `Union[Tensor, Module]`.
151
- W_ , layernorm .bias .to (dtype = torch .float32 )
151
+ W_ ,
152
+ layernorm .bias .to (dtype = torch .float32 ),
152
153
)
153
154
linear .bias .data = linear .bias .data .to (linear_dtype )
154
155
Original file line number Diff line number Diff line change @@ -1635,7 +1635,8 @@ def plan(self) -> ExecutionPlan:
1635
1635
# empty list.
1636
1636
non_const_buffer_sizes = typing .cast (
1637
1637
# 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" ],
1639
1640
),
1640
1641
container_meta_type = self .container_meta_type ,
1641
1642
)
You can’t perform that action at this time.
0 commit comments