# SASOK Model

This package contains the SASOK cognitive model architecture implemented in PyTorch, ready for use with HuggingFace Transformers.

## Structure

- `model.py` — SASOK transformer with BatchNorm + LayerNorm
- `tokenizer.py` — HuggingFace-compatible tokenizer
- `train.py` — Training script using `Trainer`
- `cli.py` — CLI for generation

## Usage

### Train
```bash
python train.py
```

### CLI Inference
```bash
python cli.py "Hello, who are you?"
```

### Push to HuggingFace Hub
```bash
from transformers import AutoModel, AutoTokenizer

model.push_to_hub("sasok-model")
tokenizer.push_to_hub("sasok-model")
```