This model is a fine-tuned version of Salesforce/codet5-small on the combined datasets cited below, plus synthetically generated data based on the referenced datasets.

Dataset Citations

  • westenfelder/NL2SH-ALFA (MIT). Westenfelder et al., "LLM-Supported Natural Language to Bash Translation", NAACL 2025. DOI: 10.18653/v1/2025.naacl-long.555.
  • tldr-pages/tldr (CC BY 4.0). Copyright the tldr-pages team and contributors. Content was modified and normalized for training; pinned at commit 9772284fdecc17e1e72a671a773460b96ac75078. License: CC BY 4.0.
  • magnumresearchgroup/bash_gen (ChatGPT-generated bash). Fu et al., "A Transformer-based Approach for Translating Natural Language to Bash Commands" and "NL2CMD: An Updated Workflow for Natural Language to Bash Commands Translation".

Intended uses

The intent of this model is to translate natural language into bash commands, primarily for linux as of now. Use with caution, it WILL happily issue destructive commands if you let it.

Framework versions

  • transformers 4.57.6
  • torch 2.13.0+cu130
  • optimum 2.1.0
  • optimum-onnx 0.1.0
  • onnxruntime 1.28.0
  • onnx 1.22.0
Downloads last month
35
Safetensors
Model size
60.5M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for diethylene/codet5-small-nl2cmd

Finetuned
(92)
this model
Finetunes
1 model