Instructions to use BAAI/AREX-Base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use BAAI/AREX-Base with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="BAAI/AREX-Base")
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 AutoProcessor, AutoModelForMultimodalLM

processor = AutoProcessor.from_pretrained("BAAI/AREX-Base")
model = AutoModelForMultimodalLM.from_pretrained("BAAI/AREX-Base", device_map="auto")
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?"}
        ]
    },
]
inputs = processor.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use BAAI/AREX-Base with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "BAAI/AREX-Base"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "BAAI/AREX-Base",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/BAAI/AREX-Base

SGLang

How to use BAAI/AREX-Base 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 "BAAI/AREX-Base" \
    --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": "BAAI/AREX-Base",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

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 "BAAI/AREX-Base" \
        --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": "BAAI/AREX-Base",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use BAAI/AREX-Base with Docker Model Runner:
```
docker model run hf.co/BAAI/AREX-Base
```

cfli commited on 6 days ago

Commit

aaeb2cf

verified ·

1 Parent(s): c29257f

Add files using upload-large-folder tool

Browse files

Files changed (29) hide show

model-00002-of-00061.safetensors +3 -0
model-00003-of-00061.safetensors +3 -0
model-00004-of-00061.safetensors +3 -0
model-00007-of-00061.safetensors +3 -0
model-00008-of-00061.safetensors +3 -0
model-00009-of-00061.safetensors +3 -0
model-00012-of-00061.safetensors +3 -0
model-00013-of-00061.safetensors +3 -0
model-00014-of-00061.safetensors +3 -0
model-00017-of-00061.safetensors +3 -0
model-00018-of-00061.safetensors +3 -0
model-00019-of-00061.safetensors +3 -0
model-00022-of-00061.safetensors +3 -0
model-00023-of-00061.safetensors +3 -0
model-00024-of-00061.safetensors +3 -0
model-00027-of-00061.safetensors +3 -0
model-00028-of-00061.safetensors +3 -0
model-00034-of-00061.safetensors +3 -0
model-00038-of-00061.safetensors +3 -0
model-00042-of-00061.safetensors +3 -0
model-00043-of-00061.safetensors +3 -0
model-00044-of-00061.safetensors +3 -0
model-00047-of-00061.safetensors +3 -0
model-00049-of-00061.safetensors +3 -0
model-00052-of-00061.safetensors +3 -0
model-00053-of-00061.safetensors +3 -0
model-00054-of-00061.safetensors +3 -0
model-00058-of-00061.safetensors +3 -0
model-00059-of-00061.safetensors +3 -0

model-00002-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:a5e2cdc957eb535ec06e3bab5187d71a5075b5a9f3ecc915a8c1f60fe37cb098
+size 4958560176

model-00003-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:2276d15158330a6e990829452ca908bebc8b2a6672f1f8f48daadd25a3a3d60b
+size 4978134472

model-00004-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:ea719f124c6281d922588f41cc93daaa33563dec7c51b2a6d911620f15426bac
+size 4953854616

model-00007-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:610da3d37cdb0bb624c3956c040942191ab14b9650f16a35950d5adba085f878
+size 4958560176

model-00008-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:064ca0fd7e312343bfb973d34afbff55769873bd938186f1130b1b5628422cd8
+size 4978134472

model-00009-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:73335623eae6de82e5ad0454a49f9a8bbbce07aab4336146c199217f6163ce04
+size 4953854616

model-00012-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:805701c4b9c428b8a2f836b43ac10ce6a259ecca00999e577ede8f2f35e86aef
+size 4958560176

model-00013-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:551bf58025b73c3311d4ae7438bfa9756288b9616edf0f5b1f9f6e277a9ffc8e
+size 4978134480

model-00014-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:a8f0e790e34ba12948d99e5b7cbc419497f5d4ebabc3167c3c240b29d312662f
+size 4953854632

model-00017-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:10df43e3dc1456937ba9a0a090088e6e66a030b6be6032317add2e9e4610a2fa
+size 4958560184

model-00018-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:dd95a19d39d23830c49580adfcfb7d756b719932ba79b3b8b4439bffd7507bd1
+size 4978134480

model-00019-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:5306760072b1f9dcc5187758fe3d29fd67b8a9486ba05426cdc8b35be16f8e61
+size 4953854632

model-00022-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:60d22d0ecf3aa0018b4217f20db8b5e036d4ba587eb6bcc0dc305fa42e5489c8
+size 4958560184

model-00023-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:dc12bcaa61f7cac84c5886e658332293eef63a4c99809c246f01c20c1e59177d
+size 4978134480

model-00024-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e7a039ef17b1737db465ee26bcd39ffc15cd70021e9e1c75345a5485c1291c1e
+size 4953854632

model-00027-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:f47bdfae9f5b45202169daf37a1728ba0ba6dad84ca57b0601f880c48246fc0f
+size 4958560184

model-00028-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:7eea4ad4f91c05a907c37b12df0d91fe1ffd26c4818b4ff4e3feeca2381c195d
+size 4978134480

model-00034-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:edc037376500ca9096d304c4f89ffc22fe25e59366994e8ea44a34d8e68ceff8
+size 4953854632

model-00038-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:9804dc89f6ab08b7d5852bedab67998e145678c7a469734e7e63464b939500af
+size 4978134480

model-00042-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:619eb3462b2158dd30810582d673d7ebe436ac6536eec47dd875cb2b57e7839d
+size 4958560184

model-00043-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:3d3f43aae53a6c93fe244e9abbf2206fcf4a810358bb17049ea4829da0a3c984
+size 4978134480

model-00044-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:556774a1c0c340f3ee0bf0c4223865a7abdec2adbacdcd6336b546460fd68864
+size 4953854632

model-00047-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:7ff79797675130eddcf95f345bab4eabd93d41cc16844b8bc82fea072dc58084
+size 4958560184

model-00049-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:c5aae780e7ce4548ac89d2a969e1e4176d0bfe2d00358f5c5207efe5686946b3
+size 4953854632

model-00052-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:24a77f28d26a5ede9bbc29cdc061aef05d500eb408c088c3698e90c1de1cb8c5
+size 4958560184

model-00053-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:04d650bd4595ab55a7954a108e7f84285d18fd5a25419cadc41dc9e5b5b8f0e3
+size 4978134480

model-00054-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:61afef3514c2a2d7a9210f7ce555cd97df1419316b3bca47cdb56932fc6cc47a
+size 4953854632

model-00058-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:6b71992db839b002cc45e8319e4d56c7eb47c145cd05284571b7fbfa674919d7
+size 4978134480

model-00059-of-00061.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:b5f42c1b61bc9ee00a79f484cbe9e1ec364025929fb8677f8b1331a553341d96
+size 4953854632