mirror of
https://github.com/zenlm/zen-identity.git
synced 2026-10-30 14:57:57 +00:00
zeekay
63ce2c4931
- Update pyproject.toml URLs to zen-identity repo - Fix ZEN_MODELS → ZEN_CODER_MODELS imports in benchmark.py and trainer.py - Simplify train_4b.py to match pattern of flash/max scripts - Rewrite train_4b_simple.py for MLX-only training - Remove hardcoded local paths, use HuggingFace dataset zenlm/zen-identity - Clean up __init__.py documentation
26 lines
659 B
Python
26 lines
659 B
Python
"""
|
|
Zen Trainer - Fine-tune Zen Coder models
|
|
|
|
Models:
|
|
zen-coder-4b Qwen3-Coder-4B 4B 8GB VRAM
|
|
zen-coder Devstral-Small-2-24B 24B 24GB VRAM
|
|
zen-coder-flash GLM-4.7-Flash 31B MoE (3B active) 24GB VRAM ⭐
|
|
zen-coder-max Kimi-K2 671B MoE (14B active) 160GB VRAM
|
|
|
|
Usage:
|
|
from zen_trainer import ZenTrainer, get_model_config
|
|
|
|
trainer = ZenTrainer("zen-coder-flash", "zenlm/zen-identity")
|
|
trainer.train()
|
|
"""
|
|
|
|
__version__ = "0.1.0"
|
|
|
|
from .models import ZEN_CODER_MODELS, get_model_config, list_models
|
|
|
|
__all__ = [
|
|
"ZEN_CODER_MODELS",
|
|
"get_model_config",
|
|
"list_models",
|
|
]
|