Downloads · 30 days
0
anlp-a2-part2-lion
neemon/anlp-a2-part2-lion
anlp-a2-part2-lion is a translation model from neemon. Use it when you need text moved from one language to another. It is set up for pytorch. The card lists the license as mit.
Decoder-only transformer with Lion (memory-efficient sign momentum), implemented from scratch.
Access
Public
Updated Sep 26, 2026
Repo size
166 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt166 MB · 98%
At a glance
- Task
- Translation
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Sep 14, 2026
- Updated
- Sep 26, 2026
- SHA
- ecda1ee8
Try a prompt
Model card
From the Hugging Face model README
ANLP Assignment 2 (part2): lion
Decoder-only transformer with Lion (memory-efficient sign momentum), implemented from scratch.
Trained from scratch for IIIT-H Advanced NLP, Monsoon 2026. Architecture, training code and evaluation live in the assignment repository. This repository holds the checkpoint and the tokenizer it was trained with, which every model in the assignment shares.
Architecture
| Setting | Value |
|---|---|
d_model | 512 |
n_layers | 8 |
n_heads | 8 |
n_kv_heads | 8 |
n_ctx | 256 |
vocab_size | 32000 |
ffn | dense |
d_ff | 2048 |
n_routed_experts | 0 |
n_shared_experts | 0 |
top_k | 0 |
norm | rmsnorm |
Training
- Scored target tokens: 43,021,354
- Best validation loss: 3.8859
- Best validation perplexity: nan
Loading
import torch
payload = torch.load("model.pt", map_location="cpu", weights_only=False)
state_dict, config = payload["model"], payload["config"]
from tokenizers import Tokenizer
tokenizer = Tokenizer.from_file("tokenizer.json")
- Task
- Translation
- Library
- pytorch
- License
- mit
- Languages
- en, vi, ja
- Created
- Sep 14, 2026
- Updated
- Sep 26, 2026