jasonacox/jojo-124M is a GPT-style language model trained using the Jojo LLM training framework. This model was fine-tuned on the TinyStoriesV2 dataset and is designed for text generation tasks.

Model Details

Model Type: GPT-style Transformer Language Model
Training Framework: Jojo LLM
Language: English
License: MIT

Architecture

Layers: 12
Hidden Size: 768
Attention Heads: 12
Context Length: 1024 tokens
Vocabulary Size: 50,304 tokens
Total Parameters: 219.6M

Training Details

Training Data

The model was trained on the TinyStoriesV2 dataset.

Training Procedure

Training Framework: Jojo LLM v2.1.0
PyTorch Version: 2.7.1+cu126
Training Device: cuda:1
Precision: bfloat16

Training Hyperparameters

Batch Size: 12
Gradient Accumulation Steps: 40
Learning Rate: 0.0006
Weight Decay: 0.1
Dropout: 0.2
Gradient Clipping: 1.0

Training Results

Usage

Using with Transformers

python

from transformers import GPT2LMHeadModel, GPT2Tokenizer

# Load model and tokenizer
model = GPT2LMHeadModel.from_pretrained("jasonacox/jojo-124M")
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")

# Generate text
input_text = "Your prompt here"
inputs = tokenizer.encode(input_text, return_tensors="pt")
outputs = model.generate(inputs, max_length=100, num_return_sequences=1, temperature=0.7)
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(generated_text)

Using with Jojo LLM

bash

# Generate text using the original Jojo framework
python gen.py jojo-124M.pt --prompt "Your prompt here"

Technical Specifications

Model Format: PyTorch
Precision: bfloat16
Framework Compatibility:
- ✅ Hugging Face Transformers
- ✅ Jojo LLM
- ✅ PyTorch

Model Card Authors

This model card was automatically generated by the Jojo LLM Hugging Face upload script.

Citation

If you use this model, please cite:

bibtex

@misc{jasonacox/jojo_124m,
  title={jasonacox/jojo-124M},
  author={Jason A. Cox},
  year={2025},
  howpublished={\url{https://github.com/jasonacox/jojo}},
  note={Trained using Jojo LLM framework}
}

Framework Information

Jojo LLM Version: 2.1.0
Generation Date: 2025-07-10T22:23:12.633549
Checkpoint: jojo-124M.pt

For more information about the Jojo LLM framework, visit: https://github.com/jasonacox/jojo