PR #17899 Move logic into pixelshuffle layer

Move logic into pixelshuffle layer #17899

amyeroberts merged 3 commits into huggingface:main from amyeroberts:move-logic-into-pixelshuffle-layer

Move all pixelshuffle logic into layer

9039899e

Rename layer

634b9f8f

Use correct input to function

2c57cfa5

amyeroberts requested a review from

sgugger 3 years ago

amyeroberts removed review request from

sgugger 3 years ago

amyeroberts requested a review from

Rocketknight1 3 years ago

amyeroberts requested a review from

sgugger 3 years ago

sgugger approved these changes on 2022-06-27

Rocketknight1 approved these changes on 2022-06-28

amyeroberts merged f71895a6 into main 3 years ago

amyeroberts deleted the move-logic-into-pixelshuffle-layer branch 3 years ago

Reviewers

Rocketknight1

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone