elisbanpaco commited on
Commit
e57d65a
·
verified ·
1 Parent(s): 4cd5153

add README

Browse files
Files changed (1) hide show
  1. README.md +38 -0
README.md ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - code
4
+ - en
5
+ tags:
6
+ - pytorch
7
+ - causal-lm
8
+ - code-generation
9
+ - custom-architecture
10
+ license: mit
11
+ ---
12
+
13
+ # NanoCode-GPT 🚀
14
+
15
+ A custom Transformer Decoder-only language model trained from scratch for Python code generation. This model was built as an educational and experimental project, featuring modern architectural choices like Rotary Position Embeddings (RoPE) and SwiGLU activations.
16
+
17
+ ## Model Details
18
+
19
+ * **Architecture:** Custom Transformer (Decoder-only)
20
+ * **Parameters:** ~57.3M
21
+ * **Context Length:** 512 tokens
22
+ * **Vocabulary Size:** 32,000 (Custom SentencePiece BPE)
23
+ * **Format:** Safetensors
24
+
25
+ ## Training Data
26
+
27
+ The model was trained on a curated dataset of ~20,000 high-quality code examples, including:
28
+ * **CodeAlpaca**: Instruction-to-code examples.
29
+ * **CodeSearchNet**: Real-world Python code with docstrings.
30
+ * **Synthetic Data**: Curated algorithmic examples (sorting, searching, data structures).
31
+
32
+ ## How to Use
33
+
34
+ Since this is a custom architecture, it requires the original `CodeGPT` class definition and inference script to run. We recommend downloading the standalone `.zip` package from the original repository or using the provided inference code to load the Safetensors weights.
35
+
36
+ ## Limitations
37
+
38
+ This is a small-scale, experimental model (~57M parameters) trained for educational purposes. It is best suited for short, simple Python functions and algorithmic tasks. It may produce syntactically incorrect code or hallucinate logic on complex requests. Always review and test the generated code.