Skip to content

Feat: Onbaord PlamoForCausalLM Architecture #351

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 8 commits into
base: main
Choose a base branch
from

Conversation

quic-shagun
Copy link
Contributor

  1. Add support for PlamoForCausalLM architecture based models
  2. Tested with Batch Size > 1
  3. As the model is not available in HF Transformers yet, Used ModuleMethodMapperTransform to map methods

@quic-amitraj quic-amitraj marked this pull request as draft April 11, 2025 04:39
@quic-shagun quic-shagun marked this pull request as ready for review April 11, 2025 08:12
@quic-amitraj
Copy link
Contributor

Please refer the PR #373 and make change in modelling file accordingly.

@quic-amitraj quic-amitraj self-requested a review April 22, 2025 05:47
@quic-amitraj quic-amitraj marked this pull request as draft April 22, 2025 06:11
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants