Skip to content

Added attn and mlp bias#84

Closed
JRosenkranz wants to merge 6 commits intomainfrom
added_attn_mlp_bias
Closed

Added attn and mlp bias#84
JRosenkranz wants to merge 6 commits intomainfrom
added_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>
…text-generation-inference-server into added_attn_mlp_bias

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
@JRosenkranz JRosenkranz requested review from njhill and tdoublep May 6, 2024 15:25
@JRosenkranz JRosenkranz self-assigned this May 6, 2024
tdoublep and others added 3 commits May 6, 2024 11:31
#### Motivation

Torch 2.3.0 was released last week and fms-extras was pulling it in as a
build dependency, leading to symbol conflicts with torch 2.2.1 (what we
are using currently). I've changed the versioning inside fms-extras to
ensure this doesn't happen.

#### Modifications

Just bumping version for fms-extras

#### Result

Resolves ImportErrors that are present with current main:
```
 ImportError: /opt/tgis/lib/python3.11/site-packages/fms_extras/paged_c.cpython-311-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda14ExchangeDeviceEa
 ```

#### Related Issues

n/a

Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
…eration-inference into added_attn_mlp_bias

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
@JRosenkranz JRosenkranz closed this May 6, 2024
Xaenalt pushed a commit to Xaenalt/text-generation-inference that referenced this pull request Sep 16, 2024
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.

2 participants