Fix model for vLLM
Browse filesThis model is currently unusable in vLLM.
This PR updates `architectures` to match https://github.com/huggingface/transformers/blob/513f76853b00bb30ee6d152d689c0e7f5359e55f/src/transformers/models/ernie4_5/modeling_ernie4_5.py#L402
- config.json +1 -1
config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"architectures": [
|
| 3 |
-
"
|
| 4 |
],
|
| 5 |
"bos_token_id": 1,
|
| 6 |
"eos_token_id": 2,
|
|
|
|
| 1 |
{
|
| 2 |
"architectures": [
|
| 3 |
+
"Ernie4_5ForCausalLM"
|
| 4 |
],
|
| 5 |
"bos_token_id": 1,
|
| 6 |
"eos_token_id": 2,
|