diffusers
refactor rotary embedding 3: so it is not on cpu
#9307

Merged

refactor rotary embedding 3: so it is not on cpu #9307

sayakpaul merged 2 commits into main from fix-torch-rope

change get_1d_rotary to accept pos as torch tensors

1ff6c42a

yiyixuxu changed the title ~~refactor rotary embedding so it is not on cpu~~ refactor rotary embedding 3: so it is not on cpu 1 year ago

Merge branch 'main' into fix-torch-rope

aa0c11f4

yiyixuxu requested a review from

sayakpaul 1 year ago

sayakpaul approved these changes on 2024-08-29

sayakpaul merged 61d96c3a into main 1 year ago

sayakpaul deleted the fix-torch-rope branch 1 year ago

cpuhrsch commented on 2024-08-29

cpuhrsch commented on 2024-08-30

Reviewers

sayakpaul

cpuhrsch

Assignees

No one assigned

Labels

None yet

Milestone

No milestone