Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Add MambaForSequenceClassification #29552

Closed
wants to merge 5 commits into from
Closed

Add MambaForSequenceClassification #29552

wants to merge 5 commits into from

Commits on Mar 23, 2024

  1. Add MambaForSequenceClassification

    @mjschock
    mjschock committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    2ed63ce View commit details
    Browse the repository at this point in the history

  2. Update docs and tests for MambaForSequenceClassification

    @mjschock
    mjschock committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    46dec9c View commit details
    Browse the repository at this point in the history

  3. Add hidden_dropout_prob to MambaConfig and pull sequence_labels from … 

    …prepare_config_and_inputs
    @mjschock
    mjschock committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    87d55c4 View commit details
    Browse the repository at this point in the history

  4. Freeze base model params for MambaForSequenceClassification and init … 

    …classifier head linear layer weights
    @mjschock
    mjschock committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    94f7a1e View commit details
    Browse the repository at this point in the history

  5. Attempt at simplifying and conforming to GPT2/GPTNeoX style

    @mjschock
    mjschock committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    b567c2b View commit details
    Browse the repository at this point in the history