Commit
•
ff07dc0
1
Parent(s):
d269012
Add attention_bias to make TGI work (#64)
Browse files- Add attention_bias to make TGI work (e774838e2163e5770175ec0e8458adc512cbabf2)
- Update config.json (ee4f33cf3f207988dd34ccf2173c916462115d20)
Co-authored-by: Philipp Schmid <[email protected]>
- config.json +1 -0
config.json
CHANGED
@@ -31,5 +31,6 @@
|
|
31 |
"torch_dtype": "bfloat16",
|
32 |
"transformers_version": "4.39.3",
|
33 |
"use_cache": true,
|
|
|
34 |
"vocab_size": 32064
|
35 |
}
|
|
|
31 |
"torch_dtype": "bfloat16",
|
32 |
"transformers_version": "4.39.3",
|
33 |
"use_cache": true,
|
34 |
+
"attention_bias": false,
|
35 |
"vocab_size": 32064
|
36 |
}
|