CoolFace
Modelpublic

Pluto-AI-Labs/Pluto-Genesis-0.6B

sourceHugging Faceapache-2.0updated 2mo agoView on Hugging Face
1likes43downloads
Model Card

<div align="center">

๐Ÿช Pluto-Genesis-0.6B

<p align="center"> <img src="./banner.png" width="100%"> </p>

An instruction-tuned sub-1B language model fine-tuned on 80K curated samples

![HuggingFace](https://huggingface.co/Siddh07ETH/Pluto-Genesis-0.6B) ![License](https://opensource.org/licenses/Apache-2.0) ![Model Size]() ![Base Model](https://huggingface.co/Qwen/Qwen3-0.6B) ![DOI](https://doi.org/10.5281/zenodo.21368749)

</div>


Model Description

Pluto-Genesis-0.6B is a fine-tuned instruction-following language model built on top of Qwen3-0.6B. It was trained using QLoRA (Quantized Low-Rank Adaptation) on a curated mixture of 80,000 high-quality instruction-response pairs spanning general reasoning, mathematical problem solving, and code generation.

This model is part of the Pluto AI research project by Siddharth N.R., exploring efficient fine-tuning of sub-1B language models on consumer-grade hardware.

Research Goal: Demonstrate that a sub-1B model fine-tuned on carefully curated data can achieve competitive performance on reasoning, math, and coding benchmarks while remaining deployable on consumer hardware.

Training Details

PropertyValue
Base ModelQwen/Qwen3-0.6B
Parameters596 Million (596,049,920)
MethodQLoRA (4-bit NF4 + LoRA)
LoRA Rankr=64, ฮฑ=128
LoRA Target Modulesqproj, kproj, vproj, oproj, gateproj, upproj, down_proj
Training Steps2,475
Final Training Loss0.2741
PrecisionFP16
OptimizerPaged AdamW 8-bit
Learning Rate2e-4 (cosine schedule)
Effective Batch Size32 (2 ร— 16 grad accum)
Sequence Length1024 tokens
HardwareTesla T4 (16 GB)
FrameworkTransformers + PEFT + TRL

Training Data

DomainDatasetSamplesSkills Targeted
๐Ÿง  General ReasoningOpenHermes-2.530,000Instruction following, reasoning
๐Ÿ”ข MathematicsOrca-Math-200K30,000Word problems, step-by-step math
๐Ÿ’ป CodeCodeFeedback-Filtered20,000Code generation, debugging
Total80,000

Benchmarks

๐Ÿ“Š Benchmarks will be added shortly. The model is currently being evaluated on ARC-Challenge, HellaSwag, MMLU, GSM8K, and TruthfulQA using lm-evaluation-harness v0.4.4.

<p align="center"> <img src="./detailed_benchmark.png" width="100%"> </p>

<p align="center"> <img src="./benchmark-comparison.png" width="100%"> </p>


Usage

GGUF Quantizations

GGUF versions for llama.cpp, Ollama and LM Studio are available here:

https://huggingface.co/Siddh07ETH/Pluto-Genesis-0.6B-GGUF

Basic Inference

python
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model = AutoModelForCausalLM.from_pretrained(
    "Siddh07ETH/Pluto-Genesis-0.6B",
    torch_dtype=torch.float16,
    device_map="auto",
)
tokenizer = AutoTokenizer.from_pretrained("Siddh07ETH/Pluto-Genesis-0.6B")

messages = [{"role": "user", "content": "Explain what a neural network is."}]

text = tokenizer.apply_chat_template(
    messages, tokenize=False, add_generation_prompt=True
)
inputs = tokenizer(text, return_tensors="pt").to(model.device)

with torch.no_grad():
    output = model.generate(
        **inputs,
        max_new_tokens=256,
        temperature=0.3,
        do_sample=True,
        top_p=0.9,
        repetition_penalty=1.1,
    )

response = tokenizer.decode(
    output[0][inputs.input_ids.shape[1]:],
    skip_special_tokens=True
)
print(response)

Low Memory Inference (4-bit)

python
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
import torch

quant_config = BitsAndBytesConfig(
    load_in_4bit=True,
    bnb_4bit_quant_type="nf4",
    bnb_4bit_compute_dtype=torch.float16,
)
model = AutoModelForCausalLM.from_pretrained(
    "Siddh07ETH/Pluto-Genesis-0.6B",
    quantization_config=quant_config,
    device_map="auto",
)
tokenizer = AutoTokenizer.from_pretrained("Siddh07ETH/Pluto-Genesis-0.6B")

Recommended Generation Settings

SettingValueReason
temperature0.3Conservative โ€” reduces hallucinations
top_p0.9Focused vocabulary
repetition_penalty1.1Prevents rambling
max_new_tokens200โ€“512Keeps answers concise
do_sampleTrueRequired when temperature < 1.0

Limitations

  • โ€”Model size: At 596M parameters this model will hallucinate on topics outside its training distribution. Always verify factual claims.
  • โ€”Context length: Trained on sequences up to 1024 tokens. Performance may degrade on longer contexts.
  • โ€”Knowledge cutoff: The model does not have access to real-time information.
  • โ€”Research only: Not intended for production deployment without further evaluation and safety testing.

Author

Siddharth N.R. HuggingFace


๐Ÿ“„ Research Paper

The complete research paper describing the Pluto-Genesis training pipeline, benchmark evaluation, checkpoint recovery system, and engineering methodology is available on Zenodo.

Paper

Pluto-Genesis: Reliable QLoRA Fine-Tuning on Ephemeral Compute with Cross-Session Checkpoint Recovery for Sub-1B Language Models

DOI https://doi.org/10.5281/zenodo.21368749

This paper includes:

  • โ€”Training methodology
  • โ€”QLoRA configuration
  • โ€”Cross-session checkpoint recovery
  • โ€”Benchmark evaluation
  • โ€”Deployment workflow
  • โ€”Reproducibility details

Citation

bibtex
@misc{plutogenesis2026,
  author    = {Siddharth N.R.},
  title     = {Pluto-Genesis-0.6B: An Instruction-Tuned Sub-1B Language Model},
  year      = {2026},
  publisher = {HuggingFace},
  url       = {https://huggingface.co/Siddh07ETH/Pluto-Genesis-0.6B}
}

License

Apache 2.0 โ€” see LICENSE. Base model Qwen3-0.6B is also Apache 2.0.