CoolFace
Modelpublic

sathishphdai/uiux-designer-slm-5m

sourceHugging Facemitupdated 7mo agoView on Hugging Face
1likes9downloads
Model Card

UI/UX Designer-SLM: Role-Based Small Language Model

A LLaMA-style transformer (~989.3M params, ~0.99B) trained from scratch for the UI/UX Designer role. Supports up to 5M token context via RoPE with gradient checkpointing.

Architecture

ComponentValue
ArchitectureLLaMA-style (RoPE + RMSNorm + SwiGLU)
Parameters~989.3M (~0.99B)
Layers32
Heads20
Embedding1600
Max Context5,000,000 tokens
Max Output5,000,000 tokens
Vocab1,805 BPE
Model Size~4 GB (fp32)

Training

  • —Best eval loss: 3.3150162220001222
  • —Trained with gradient checkpointing on Apple M4 (MPS)
  • —3 epochs, batchsize=1, gradaccum=16

Usage

python
from huggingface_hub import hf_hub_download
from tokenizers import Tokenizer

model_path = hf_hub_download("sathishphdai/uiux-designer-slm-5m", "model.safetensors")
tokenizer_path = hf_hub_download("sathishphdai/uiux-designer-slm-5m", "uiux_designer_tokenizer.json")
tokenizer = Tokenizer.from_file(tokenizer_path)