Image-Text-to-Text
Transformers
Safetensors
English
Chinese
ernie4_5_moe_vl
ERNIE4.5
conversational
custom_code
Instructions to use baidu/ERNIE-4.5-VL-28B-A3B-PT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use baidu/ERNIE-4.5-VL-28B-A3B-PT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-text-to-text", model="baidu/ERNIE-4.5-VL-28B-A3B-PT", trust_remote_code=True) messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] pipe(text=messages)# Load model directly from transformers import AutoModelForImageTextToText model = AutoModelForImageTextToText.from_pretrained("baidu/ERNIE-4.5-VL-28B-A3B-PT", trust_remote_code=True, dtype="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use baidu/ERNIE-4.5-VL-28B-A3B-PT with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "baidu/ERNIE-4.5-VL-28B-A3B-PT" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "baidu/ERNIE-4.5-VL-28B-A3B-PT", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker
docker model run hf.co/baidu/ERNIE-4.5-VL-28B-A3B-PT
- SGLang
How to use baidu/ERNIE-4.5-VL-28B-A3B-PT with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "baidu/ERNIE-4.5-VL-28B-A3B-PT" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "baidu/ERNIE-4.5-VL-28B-A3B-PT", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "baidu/ERNIE-4.5-VL-28B-A3B-PT" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "baidu/ERNIE-4.5-VL-28B-A3B-PT", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }' - Docker Model Runner
How to use baidu/ERNIE-4.5-VL-28B-A3B-PT with Docker Model Runner:
docker model run hf.co/baidu/ERNIE-4.5-VL-28B-A3B-PT
Update configuration_ernie4_5_vl.py
Browse files- configuration_ernie4_5_vl.py +13 -13
configuration_ernie4_5_vl.py
CHANGED
|
@@ -539,19 +539,19 @@ class Ernie4_5_VLMoEConfig(Ernie4_5_MoEConfig):
|
|
| 539 |
"activation_function": "hidden_act",
|
| 540 |
}
|
| 541 |
base_model_tp_plan = {
|
| 542 |
-
"
|
| 543 |
-
"
|
| 544 |
-
"
|
| 545 |
-
"
|
| 546 |
-
"
|
| 547 |
-
"
|
| 548 |
-
"
|
| 549 |
-
"
|
| 550 |
-
"
|
| 551 |
-
"
|
| 552 |
-
"
|
| 553 |
-
"
|
| 554 |
-
"
|
| 555 |
}
|
| 556 |
|
| 557 |
def __init__(
|
|
|
|
| 539 |
"activation_function": "hidden_act",
|
| 540 |
}
|
| 541 |
base_model_tp_plan = {
|
| 542 |
+
"model.layers.*.self_attn.q_proj": "colwise_rep",
|
| 543 |
+
"model.layers.*.self_attn.k_proj": "colwise_rep",
|
| 544 |
+
"model.layers.*.self_attn.v_proj": "colwise_rep",
|
| 545 |
+
"model.layers.*.self_attn.o_proj": "rowwise_rep",
|
| 546 |
+
"model.layers.*.mlp.experts.*.gate_proj": "colwise",
|
| 547 |
+
"model.layers.*.mlp.experts.*.up_proj": "colwise",
|
| 548 |
+
"model.layers.*.mlp.experts.*.down_proj": "rowwise",
|
| 549 |
+
"model.layers.*.mlp_text.experts.*.gate_proj": "colwise",
|
| 550 |
+
"model.layers.*.mlp_text.experts.*.up_proj": "colwise",
|
| 551 |
+
"model.layers.*.mlp_text.experts.*.down_proj": "rowwise",
|
| 552 |
+
"model.layers.*.mlp.gate_proj": "colwise",
|
| 553 |
+
"model.layers.*.mlp.up_proj": "colwise",
|
| 554 |
+
"model.layers.*.mlp.down_proj": "rowwise"
|
| 555 |
}
|
| 556 |
|
| 557 |
def __init__(
|