Instructions to use XYZAILab/XYZ-Aquila-mini with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use XYZAILab/XYZ-Aquila-mini with Transformers:

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

pipe = pipeline("text-generation", model="XYZAILab/XYZ-Aquila-mini")
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("XYZAILab/XYZ-Aquila-mini")
model = AutoModelForMultimodalLM.from_pretrained("XYZAILab/XYZ-Aquila-mini", 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 XYZAILab/XYZ-Aquila-mini with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/XYZAILab/XYZ-Aquila-mini

SGLang

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

Docker Model Runner
How to use XYZAILab/XYZ-Aquila-mini with Docker Model Runner:
```
docker model run hf.co/XYZAILab/XYZ-Aquila-mini
```

search_web versus web_search

by kbuettner - opened Aug 11

Discussion

kbuettner

Aug 11

XYZAILab/XYZ-Aquila-mini wants to search the web (only) using web_search, a non-existent tool for Open WebUI. When using Open WebUI, the correct tool to use is search_web; the model is able to list it when I tell it to show me its available tool calls. When I tell it to use search_web instead of web_search, it tries repeatedly, but always ends up calling web_search instead.

bardd

about 1 month ago

So change your tool name, and help the model a bit, at the end it's just a word calculator

XYZAILab

Owner 27 days ago

Thank you for your interest in our work!

For evaluation, we recommend using our harness available here: https://github.com/XYZ-AI-Lab/AxisAgentic

The default tool schema aligns with those used during the model's training, ensuring consistent performance during evaluation.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment