Views
No views yet
1cd Zenith/V1-Tenstorrent-Blackhole-p300/32B
2pip install -r requirements.txt1# LoRA fine-tuning (recommended)
2python train.py \
3 --base_model deepseek-ai/DeepSeek-R1-Distill-Qwen-32B \
4 --train_data ./data/train.json \
5 --use_lora \
6 --lora_r 16 \
7 --lora_alpha 32 \
8 --epochs 3 \
9 --batch_size 4 \
10 --gradient_accumulation_steps 8 \
11 --learning_rate 1e-4 \
12 --use_ring_attention \
13 --max_seq_length 32768 \
14 --tensor_parallel_size 8 \
15 --pipeline_parallel_size 4 \
16 --use_noc_optimization \
17 --mixed_precision bf161# Interactive mode
2python inference.py --checkpoint ./outputs/checkpoint-final
3
4# Single prompt
5python inference.py \
6 --checkpoint ./outputs/checkpoint-final \
7 --prompt "Write a Python function to implement quicksort" \
8 --max_new_tokens 10241ollama create zenith-32b-p300 -f Modelfile
2ollama run zenith-32b-p300 "Explain the difference between supervised and unsupervised learning"1from configs.zenith_config import get_32b_config
2
3config = get_32b_config()hidden_size: 4096num_layers: 40num_heads: 32num_experts: 8 (configurable)moe_top_k: 2max_seq_len: 32768use_ring_attention: Truering_attention_chunk_size: 8192ring_attention_overlap: 20481from data.openthoughts_processor import OpenThoughtsProcessor, OpenThoughtsConfig
2
3ot_config = OpenThoughtsConfig(
4 dataset_name="open-thoughts/OpenThoughts3-1.2M",
5 streaming=True,
6 max_seq_length=32768,
7 quality_filtering=True,
8 curriculum_learning=True,
9 tokenizer=tokenizer
10)
11processor = OpenThoughtsProcessor(ot_config)--use_moe --num_experts 8 --moe_top_k 2--use_eq_adapter --eq_loss_weight 0.05--use_ring_attention --ring_chunk_size 8192 --ring_overlap 2048python test_model.py1python -m evaluation.benchmark \
2 --model_path ./outputs/checkpoint-final \
3 --benchmarks humaneval mbpp gsm8k math truthfulqa1ollama create zenith-32b-p300 -f Modelfile
2ollama run zenith-32b-p300 "Your prompt here"1python -m vllm.entrypoints.openai.api_server \
2 --model ./outputs/checkpoint-final \
3 --tensor-parallel-size 2 \
4 --max-model-len 32768 \
5 --port 8000| Configuration | Memory | Speed | Quality |
|---|---|---|---|
| Full FT, 2K | ~58GB | 50-80 | Baseline |
| LoRA r=16, 2K | ~18GB | 80-120 | 98% |
| QLoRA r=8, 2K | ~10GB | 100-150 | 95% |
| Ring 32K | +20% | 30-50 | Enables long context |
1@misc{zenith-32b-p300-2025,
2 title={Zenith-32B-p300: A Tenstorrent-Optimized Reasoning Model},
3 author={Zenith Project},
4 year={2025}
5}README.mdFINETUNE_GUIDE.mdconfigs/zenith_config.py