Base model
Merged from google/gemma-4-31B-it and the AOS Manim SFT LoRA adapter. Accept the Gemma license and set HF_TOKEN to download.
Training data
Fine-tuned on nabin2004/AOS-Trajectories using the AOS Phase 1 SFT trainer (apps/sft).
Usage
import os
import torch
from transformers import AutoModelForImageTextToText, AutoTokenizer
token = os.environ["HF_TOKEN"]
model_id = "nabin2004/AOS-gemma4-31b-manim-merged"
tokenizer = AutoTokenizer.from_pretrained(model_id, token=token)
model = AutoModelForImageTextToText.from_pretrained(
model_id,
torch_dtype=torch.bfloat16,
device_map="auto",
token=token,
)
vLLM
Serve as a regular model (no --enable-lora needed):
vllm serve nabin2004/AOS-gemma4-31b-manim-merged --max-model-len 16384
Ollama / local GGUF
Convert to GGUF or pull the pre-built GGUF repo. See apps/sft/export_gguf.py.
How this was produced
cd apps/sft
uv run python merge_adapter.py \
--adapter-dir ./gemma4-31b-manim-ft \
--output-dir ./gemma4-31b-manim-merged \
--push-to-hub