You can deploy this model and start earning money today!

ricemonster/
IEEE-T-ASE

Views

No views yet

NeurIPS11092: Text-to-CadQuery Repo

This repository contains all resources used to train and evaluate large language models for generating CadQuery code from natural language descriptions.

data/
Contains prompt-completion pairs used to finetune six open-source LLMs.
These files are split into data_train.jsonl, data_val.jsonl, and data_test.jsonl following a 90/5/5 ratio.
CadQuery.zip
Includes all 170,000 CadQuery programs we generated from the Text2CAD dataset using Gemini 2.0 Flash.
text2cad_v1.1.csv
Original source data provided by the Text2CAD authors, in minimal JSON format.

Finetuned Models

We trained the following models on this dataset:

CodeGPT-small
GPT-2 Medium
GPT-2 Large
Gemma-1B
Qwen2.5-3B
Mistral-7B (LoRA)

Acknowledgements

We gratefully acknowledge the authors of Text2CAD and DeepCAD for their foundational datasets and inspiration.

ricemonster/IEEE-T-ASE

NeurIPS11092: Text-to-CadQuery Repo

Contents

Finetuned Models

Acknowledgements

ricemonster/
IEEE-T-ASE