Fix weights after#17785

Wrong weight size

ArthurZ changed pull request status to closed