Instructions to use mohith-das/jetson-assistant-stage1-adapter with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use mohith-das/jetson-assistant-stage1-adapter with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-0.5B") model = PeftModel.from_pretrained(base_model, "mohith-das/jetson-assistant-stage1-adapter") - Notebooks
- Google Colab
- Kaggle
jetson-assistant-stage1-adapter
Stage 1 ("cognitive distillation") LoRA adapter from the llm_distillery project โ a personal-assistant training pipeline run entirely on a Jetson Orin Nano Super (8GB unified memory), start to finish.
Qwen2.5-0.5B imitates small, high-signal slices of three teacher-grade datasets to pick up reasoning, coding, and multi-turn conversational structure:
| Source | Rows | Teaches |
|---|---|---|
a-m-team/AM-DeepSeek-R1-Distilled-1.4M |
5,000 | step-by-step reasoning traces (chain-of-thought) |
nvidia/OpenCodeInstruct (shard 0) |
2,000 | Python code Q&A, unit-test-validated |
HuggingFaceM4/FineVision (text configs) |
1,000 | multi-turn conversational structure |
Training: LoRA r=32/alpha=64 on all linear projections (q/k/v/o/gate/up/down proj), lr=2e-4, 1 epoch (~500 steps), effective batch 16 (batch_size=1 x grad_accum=16), max_length=1024, FP16.
This is the standalone Stage 1 adapter. Stage 2
(jetson-assistant-stage2-adapter)
was trained on top of this adapter merged into the base, not stacked
alongside it โ see
jetson-assistant-0.5b
for the final deployable merge of both stages.
- Downloads last month
- 18
Model tree for mohith-das/jetson-assistant-stage1-adapter
Base model
Qwen/Qwen2.5-0.5B