Instructions to use KrynexLabs/KrynexAI-1.3-50M-Mobile-Model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use KrynexLabs/KrynexAI-1.3-50M-Mobile-Model with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="KrynexLabs/KrynexAI-1.3-50M-Mobile-Model", trust_remote_code=True)

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("KrynexLabs/KrynexAI-1.3-50M-Mobile-Model", trust_remote_code=True, device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use KrynexLabs/KrynexAI-1.3-50M-Mobile-Model with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "KrynexLabs/KrynexAI-1.3-50M-Mobile-Model"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "KrynexLabs/KrynexAI-1.3-50M-Mobile-Model",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/KrynexLabs/KrynexAI-1.3-50M-Mobile-Model

SGLang

How to use KrynexLabs/KrynexAI-1.3-50M-Mobile-Model with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "KrynexLabs/KrynexAI-1.3-50M-Mobile-Model" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "KrynexLabs/KrynexAI-1.3-50M-Mobile-Model",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "KrynexLabs/KrynexAI-1.3-50M-Mobile-Model" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "KrynexLabs/KrynexAI-1.3-50M-Mobile-Model",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use KrynexLabs/KrynexAI-1.3-50M-Mobile-Model with Docker Model Runner:
```
docker model run hf.co/KrynexLabs/KrynexAI-1.3-50M-Mobile-Model
```

KrynexAI-50M

KrynexAI-50M is a 50M-parameter language model designed with a 16,384-token vocabulary and 16,384-token context window.

Model Details

Architecture: Hybrid Mamba2 / Transformer
Block Pattern: 3 Mamba2 blocks : 1 Attention block (repeating)
Parameters: ~49,334,448 (50M)
Hidden Dimension: 768
Layers: 8 (6 Mamba2, 2 Attention)
Vocab Size: 16,384 (Custom Byte-Level BPE)
Context Length: 16,384
Training Tokens: ~~25,000,000,000 (~~25 Billion)
Optimizer: Muon (for 2D hidden weights) + AdamW (for embeddings, norms, and scalars)
Precision: fp32 master weights with bf16 autocast

Dataset Sources

The model was trained on a 25B-token subset of the following datasets:

Dataset	Token Allocation	Share
FineWeb-Edu	7.50 billion	30%
DCLM	5.00 billion	20%
Cosmopedia-v2	3.75 billion	15%
FineMath-4+	3.75 billion	15%
FinePhrase	3.00 billion	12%
NPset	2.00 billion	8%
Total	25.00 billion	100%

Original author

Pebble (50M)

License

Apache 2.0

Usage

KrynexAI-50M does not require the mamba-ssm library and is intended to be usable with standard PyTorch-based inference implementations. It may run on CUDA GPUs, AMD GPUs, Intel GPUs, and CPUs depending on the inference framework and available hardware acceleration.

from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("your-username/KrynexAI-50M", trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("KeynexLabs/KrynexAI-50M-Mobile-Model")

Downloads last month: 250

Safetensors

Model size

61.9M params

Tensor type

F32

Collection including KrynexLabs/KrynexAI-1.3-50M-Mobile-Model

The Main Collection

Collection

5 items • Updated about 5 hours ago • 1