Skip to content

added mlp and attn bias option to flash and paged llama models#85

Merged
njhill merged 2 commits intomainfrom
attn_mlp_bias
May 6, 2024
Merged

added mlp and attn bias option to flash and paged llama models#85
njhill merged 2 commits intomainfrom
attn_mlp_bias

Conversation

@JRosenkranz
Copy link
Copy Markdown
Collaborator

Motivation

[Describe why this change is needed]

The Calico models currently set the mlp and attention bias to true, which was hard-coded to false in flash and paged llama implementations. This will use the config params set in huggingface/transformers#30031 to set those values properly.

Modifications

[Describe the code changes]

  • added attention_bias, mlp_bias to config for Flash and Paged Llama implementations (default is False)
  • set bias in attention and mlp to the config value

Result

[Describe how the changes affects existing behavior and how to test it]

Models should be able to load properly if containing attention and mlp bias

Related Issues

NA

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
@JRosenkranz JRosenkranz requested review from njhill and tdoublep May 6, 2024 15:36
@JRosenkranz JRosenkranz self-assigned this May 6, 2024
Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
Copy link
Copy Markdown
Contributor

@njhill njhill left a comment

Choose a reason for hiding this comment

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

@njhill njhill merged commit deb99f6 into main May 6, 2024
@njhill njhill deleted the attn_mlp_bias branch May 6, 2024 21:18
openshift-merge-bot bot referenced this pull request in red-hat-data-services/text-generation-inference May 7, 2024
added mlp and attn bias option to flash and paged llama models (#85)
Xaenalt pushed a commit to Xaenalt/text-generation-inference that referenced this pull request Aug 1, 2024
[pull] main from IBM:main
Xaenalt pushed a commit to Xaenalt/text-generation-inference that referenced this pull request Aug 7, 2024
#### Motivation

The `Calico` models currently set the mlp and attention bias to true,
which was hard-coded to false in flash and paged llama implementations.
This will use the config params set in
huggingface/transformers#30031 to set those
values properly.

#### Modifications

- added attention_bias, mlp_bias to config for Flash and Paged Llama
implementations (default is False)
- set bias in attention and mlp to the config value

#### Result

Models should be able to load properly if containing attention and mlp
bias

---------

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
Co-authored-by: Joe Runde <Joseph.Runde@ibm.com>
Xaenalt pushed a commit to Xaenalt/text-generation-inference that referenced this pull request Aug 12, 2024
#### Motivation

The `Calico` models currently set the mlp and attention bias to true,
which was hard-coded to false in flash and paged llama implementations.
This will use the config params set in
huggingface/transformers#30031 to set those
values properly.

#### Modifications

- added attention_bias, mlp_bias to config for Flash and Paged Llama
implementations (default is False)
- set bias in attention and mlp to the config value

#### Result

Models should be able to load properly if containing attention and mlp
bias

---------

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
Co-authored-by: Joe Runde <Joseph.Runde@ibm.com>
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.

3 participants