---
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.