Text Generation
Transformers
PyTorch
GGUF
code
multiscale_transformer
code-generation
multi-scale-transformer
cpu-optimized
cubic
llama
byte-level
vision
multi-modal
Eval Results (legacy)
Instructions to use CubicLabs/AXL-Vision-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use CubicLabs/AXL-Vision-v2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="CubicLabs/AXL-Vision-v2")# Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("CubicLabs/AXL-Vision-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use CubicLabs/AXL-Vision-v2 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "CubicLabs/AXL-Vision-v2" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "CubicLabs/AXL-Vision-v2", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/CubicLabs/AXL-Vision-v2
- SGLang
How to use CubicLabs/AXL-Vision-v2 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 "CubicLabs/AXL-Vision-v2" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "CubicLabs/AXL-Vision-v2", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'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 "CubicLabs/AXL-Vision-v2" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "CubicLabs/AXL-Vision-v2", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use CubicLabs/AXL-Vision-v2 with Docker Model Runner:
docker model run hf.co/CubicLabs/AXL-Vision-v2
Update README.md
Browse files
README.md
CHANGED
|
@@ -6,7 +6,7 @@ tags:
|
|
| 6 |
- code-generation
|
| 7 |
- multi-scale-transformer
|
| 8 |
- cpu-optimized
|
| 9 |
-
-
|
| 10 |
- pytorch
|
| 11 |
- llama
|
| 12 |
- gguf
|
|
@@ -16,7 +16,7 @@ tags:
|
|
| 16 |
pipeline_tag: text-generation
|
| 17 |
library_name: transformers
|
| 18 |
datasets:
|
| 19 |
-
-
|
| 20 |
model-index:
|
| 21 |
- name: AXL-Vision-v2
|
| 22 |
results:
|
|
@@ -30,13 +30,13 @@ model-index:
|
|
| 30 |
|
| 31 |
# AXL-Vision-v2
|
| 32 |
|
| 33 |
-
UI vision. 4.1M params. Screen-to-code prototype. Trained on synthetic UI screenshots. Part of the AXL model family by [
|
| 34 |
|
| 35 |
## Model Details
|
| 36 |
|
| 37 |
| Property | Value |
|
| 38 |
|----------|-------|
|
| 39 |
-
| Developed by | [
|
| 40 |
| Architecture | Multi-Scale Transformer |
|
| 41 |
| Parameters | 4M |
|
| 42 |
| Optimizer | SGD |
|
|
@@ -51,8 +51,8 @@ UI vision. 4.1M params. Screen-to-code prototype. Trained on synthetic UI screen
|
|
| 51 |
|
| 52 |
### Sources
|
| 53 |
|
| 54 |
-
- **Repository:** [GitHub](https://github.com/
|
| 55 |
-
- **Organization:** [
|
| 56 |
|
| 57 |
## Uses
|
| 58 |
|
|
@@ -152,9 +152,9 @@ Multi-Scale Transformer with three parallel encoder stacks at resolution scales
|
|
| 152 |
```bibtex
|
| 153 |
@misc{axl_2026,
|
| 154 |
title={AXL: AXL-Vision-v2 - Multi-Scale Transformer for CPU Code Generation},
|
| 155 |
-
author={
|
| 156 |
year={2026},
|
| 157 |
-
url={https://huggingface.co/
|
| 158 |
}
|
| 159 |
```
|
| 160 |
|
|
@@ -185,4 +185,4 @@ ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
|
|
| 185 |
with torch.no_grad():
|
| 186 |
out = model.generate(ids, max_new_tokens=100, temperature=0.8, top_k=40)
|
| 187 |
print(tokenizer.decode(out[0].tolist()))
|
| 188 |
-
```
|
|
|
|
| 6 |
- code-generation
|
| 7 |
- multi-scale-transformer
|
| 8 |
- cpu-optimized
|
| 9 |
+
- cubic
|
| 10 |
- pytorch
|
| 11 |
- llama
|
| 12 |
- gguf
|
|
|
|
| 16 |
pipeline_tag: text-generation
|
| 17 |
library_name: transformers
|
| 18 |
datasets:
|
| 19 |
+
- cubic/axl-synthetic-ui
|
| 20 |
model-index:
|
| 21 |
- name: AXL-Vision-v2
|
| 22 |
results:
|
|
|
|
| 30 |
|
| 31 |
# AXL-Vision-v2
|
| 32 |
|
| 33 |
+
UI vision. 4.1M params. Screen-to-code prototype. Trained on synthetic UI screenshots. Part of the AXL model family by [CubicLabs](https://huggingface.co/CubicLabs).
|
| 34 |
|
| 35 |
## Model Details
|
| 36 |
|
| 37 |
| Property | Value |
|
| 38 |
|----------|-------|
|
| 39 |
+
| Developed by | [CubicLabs](https://huggingface.co/CubicLabs) |
|
| 40 |
| Architecture | Multi-Scale Transformer |
|
| 41 |
| Parameters | 4M |
|
| 42 |
| Optimizer | SGD |
|
|
|
|
| 51 |
|
| 52 |
### Sources
|
| 53 |
|
| 54 |
+
- **Repository:** [GitHub](https://github.com/Cubic/AXL)
|
| 55 |
+
- **Organization:** [CubicLabs](https://huggingface.co/CubicLabs)
|
| 56 |
|
| 57 |
## Uses
|
| 58 |
|
|
|
|
| 152 |
```bibtex
|
| 153 |
@misc{axl_2026,
|
| 154 |
title={AXL: AXL-Vision-v2 - Multi-Scale Transformer for CPU Code Generation},
|
| 155 |
+
author={Cubic},
|
| 156 |
year={2026},
|
| 157 |
+
url={[https://huggingface.co/CubicLabs](https://huggingface.co/CubicLabs)}
|
| 158 |
}
|
| 159 |
```
|
| 160 |
|
|
|
|
| 185 |
with torch.no_grad():
|
| 186 |
out = model.generate(ids, max_new_tokens=100, temperature=0.8, top_k=40)
|
| 187 |
print(tokenizer.decode(out[0].tolist()))
|
| 188 |
+
```
|