Instructions to use soyrsoyr/gemma-4-unified-0.8B-tiny with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use soyrsoyr/gemma-4-unified-0.8B-tiny with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-text-to-text", model="soyrsoyr/gemma-4-unified-0.8B-tiny") messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] pipe(text=messages)# Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("soyrsoyr/gemma-4-unified-0.8B-tiny") model = AutoModelForMultimodalLM.from_pretrained("soyrsoyr/gemma-4-unified-0.8B-tiny", device_map="auto") messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] inputs = processor.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use soyrsoyr/gemma-4-unified-0.8B-tiny with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "soyrsoyr/gemma-4-unified-0.8B-tiny" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "soyrsoyr/gemma-4-unified-0.8B-tiny", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker
docker model run hf.co/soyrsoyr/gemma-4-unified-0.8B-tiny
- SGLang
How to use soyrsoyr/gemma-4-unified-0.8B-tiny with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "soyrsoyr/gemma-4-unified-0.8B-tiny" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "soyrsoyr/gemma-4-unified-0.8B-tiny", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "soyrsoyr/gemma-4-unified-0.8B-tiny" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "soyrsoyr/gemma-4-unified-0.8B-tiny", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }' - Docker Model Runner
How to use soyrsoyr/gemma-4-unified-0.8B-tiny with Docker Model Runner:
docker model run hf.co/soyrsoyr/gemma-4-unified-0.8B-tiny
This is a tiny version of google/gemma-4-12B-it created for testing and development.
Intended Use
A small, fast stand-in for the gemma4_unified architecture, useful for:
- Inference / CI testing where a real 12B checkpoint is too large to download or run
- Quantization & compression pipeline smoke tests (llm-compressor, compressed-tensors)
- Offloaded / distributed loading tests (see below)
- Exercising the vLLM
Gemma4UnifiedForConditionalGenerationmodel path
Weights are random (then briefly fine-tuned on a toy corpus), so generations are not meaningful, this model is for plumbing, not output quality.
Model Details
- Base Model: google/gemma-4-12B-it
- Architecture: gemma4_unified (Gemma4UnifiedForConditionalGeneration)
- Total Parameters: 0.798B
- Activated Parameters: 0.798B
Configuration Changes
The following parameters were reduced from the original model:
| Parameter | Original | Tiny |
|---|---|---|
| text_config.num_hidden_layers | 48 | 6 |
| text_config.hidden_size | 3840 | 2048 |
| text_config.intermediate_size | 15360 | 1536 |
Layer type patterns are preserved: the sliding/full attention mix is kept as
5× sliding_attention + 1× full_attention. Attention shape is unchanged
(16 heads, 8 KV heads, head_dim 256), and the full 262,144-token vocabulary is
retained. The vision (gemma4_unified_vision) and audio towers are left at base
size; they are small relative to the shared token embedding.
Checkpoint Structure
Single safetensors file (model.safetensors). Key naming matches the original
checkpoint format (model.language_model.*, model.embed_vision.*,
model.embed_audio.*, model.vision_embedder.*). Module-path structure was
verified equal to the base checkpoint's safetensors header.
Usage
from transformers import AutoModelForImageTextToText, AutoProcessor
model = AutoModelForImageTextToText.from_pretrained(
"soyrsoyr/gemma-4-unified-0.8B-tiny", device_map="auto"
)
processor = AutoProcessor.from_pretrained("soyrsoyr/gemma-4-unified-0.8B-tiny")
input_ids = processor.tokenizer("According to all known laws", return_tensors="pt").input_ids.to(model.device)
output = model.generate(input_ids, max_new_tokens=20)
print(processor.tokenizer.decode(output[0]))
Offloaded / distributed loading (compressed-tensors)
AutoModelForCausalLM resolves this architecture (transformers maps
gemma4_unified into the causal-LM auto-class), so the default
load_offloaded_model() pattern works directly:
from transformers import AutoModelForCausalLM
from compressed_tensors.offload import load_offloaded_model
from compressed_tensors.distributed import init_dist
init_dist()
with load_offloaded_model(): # patches AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained(
"soyrsoyr/gemma-4-unified-0.8B-tiny",
device_map="auto_offload", # weights on CPU/disk, GPU for activations
)
Creation Process
This model was created using the llm-compressor create-tiny-model claude skill.
- Config inspected via
inspect_config.py - Tiny model created via a modified
save_tiny_model.py, adapted for the multimodal class (AutoModelForImageTextToText.from_config); only the text tower was shrunk and any all-zero / non-finite / extreme param was fixed afterinit_weights() - Fine-tuned on the copypasta dataset; reached training perplexity 1.33 (target: ≤3.0) at lr=5e-4 (CPU, Adafactor)
- Checkpoint structure validated against the original HuggingFace safetensors header (module-path match)
- Inference validated via
validate_tiny_model.py
Notes
- Text-only shrink. The vision projector
(
embed_vision.multimodal_embedder.embedding_projection) readstext_config.hidden_sizefor its output dim, so it auto-aligns to the reduced 2048 hidden.vision_config.output_proj_dims/mm_embed_dimare intentionally left unchanged — editing them flips the modeling code to a different (flattenedvision_embedder) module tree that no longer matches the original checkpoint layout. tie_word_embeddings=True:lm_headsharesembed_tokensand is not stored as a separate tensor.
Validation output: Success: 1.0278589725494385 <= 10.0
- Downloads last month
- 97