Views
No views yet
1import torch
2
3# Load a specific model
4model_path = "TokenUnify-1B.pth" # or any other model file
5checkpoint = torch.load(model_path, map_location='cpu')
6
7# Your model loading code here| Model | Parameters | File Name |
|---|---|---|
| TokenUnify Large | 1B | TokenUnify-1B.pth |
| TokenUnify Medium | 500M | TokenUnify-500M.pth |
| TokenUnify Small | 200M | TokenUnify-200M.pth |
| TokenUnify Tiny | 100M | TokenUnify-100M.pth |
| Superhuman Baseline | - | superhuman.pth |