File size: 1,001 Bytes
138e9ad 1e145bd 138e9ad 1e145bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | ---
license: mit
language:
- en
tags:
- text-generation
- gpt
- from-scratch
- code
---
# Cortex 2 Code
A small GPT model for code generation, trained completely from scratch (custom PyTorch architecture, no `transformers` library involved).
- **Parameters**: 120m
- **Weights format**: fp32
- **Architecture**: Decoder-only Transformer (custom implementation)
- **Training**: from scratch, no pretrained weights used
- **Language**: writes Python code, all comments and output in English
> ⚠️ **Note:** the context window is only **512 tokens**, so the model will forget earlier parts of the code quickly.
> ⚠️ **Note:** this is a small model for code, so don't expect great results from it — but it can be fun to try out of curiosity.
## How to run
2. Install dependencies:
pip install -r requirements.txt
3. Run:
python chat.py
## Limitations
This is a small model, so it may produce broken, incomplete, or non-functional code — expected behavior for this parameter count. |