FutureCAD-7B

FutureCAD-7B generates CadQuery programs from text descriptions. It is based on Qwen2.5-7B-Instruct.

Paper · Code · Dataset · BRepGround

Text-to-CAD

Install the futurecad environment from the code repository. Run from that repository's root:

python scripts/download_models.py
bash scripts/serve_grounding.sh

In another terminal using the same environment:

export VLLM_USE_FLASHINFER_SAMPLER=0
python -m futurecad.text_to_cad \
  --text "Create a cylinder with radius 10 mm and height 20 mm." \
  --model weights/futurecad-llm \
  --output-dir outputs/cylinder

The command saves the generated Python program and exports STEP/STL when CAD execution succeeds. The repository supplies the prompt template and CadQuery extensions used by the model.

Load with Transformers

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "jhlee11/FutureCAD-7B"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, dtype="auto", device_map="auto")

For prompt formatting, generation, execution and evaluation, use the code repository.

Citation

@misc{li2026highfidelitycadgenerationllmdriven,
  title={Towards High-Fidelity CAD Generation via LLM-Driven Program Generation and Text-Based B-Rep Primitive Grounding},
  author={Jiahao Li and Qingwang Zhang and Qiuyu Chen and Guozhan Qiu and Yunzhong Lou and Xiangdong Zhou},
  year={2026},
  eprint={2603.11831},
  archivePrefix={arXiv},
  primaryClass={cs.CV},
  url={https://arxiv.org/abs/2603.11831},
}
Downloads last month
105
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for jhlee11/FutureCAD-7B

Base model

Qwen/Qwen2.5-7B
Finetuned
(3143)
this model
Quantizations
1 model

Dataset used to train jhlee11/FutureCAD-7B

Paper for jhlee11/FutureCAD-7B