Skip to content

liger-kernels: Add layer repos for remaining functions#995

Open
vasqu wants to merge 1 commit into
mainfrom
adjust-liger-to-all-layers
Open

liger-kernels: Add layer repos for remaining functions#995
vasqu wants to merge 1 commit into
mainfrom
adjust-liger-to-all-layers

Conversation

@vasqu

@vasqu vasqu commented Jun 25, 2026

Copy link
Copy Markdown
Collaborator

As per title, adds layer type versions for the functions

LigerForCausalLMLoss.can_torch_compile = True


class liger_rotary_pos_emb_layer(nn.Module):

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

added layer suffices as it would clash with the original fns otherwise and we should keep BC imo

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant