Views
No views yet
nvidia-modelopt. Hardware-agnostic checkpoint; inference requires NVIDIA Blackwell GPUs (RTX 50xx, B100/B200, GB200) via TensorRT-LLM.config.json # Gemma4ForCausalLM, NVFP4 quantization metadata
generation_config.json
tokenizer.json + tokenizer_config.json + chat_template.jinja
model.safetensors # NVFP4 weights (~5-6 GB)1# Download
2git lfs install
3git clone https://huggingface.co/tss-deposium/gemma-4-E4B-text-only-nvfp4
4cd gemma-4-E4B-text-only-nvfp4
5
6# Validate before the full build (~30s) — cheap signal for compatibility
7trtllm-build --checkpoint_dir . --output_dir /tmp/dryrun \
8 --dry_run --log_level debug
9
10# Full engine build (10-30 min on RTX 50xx)
11trtllm-build --checkpoint_dir . \
12 --output_dir ./engine \
13 --gemm_plugin nvfp4 \
14 --max_batch_size 4 --max_input_len 4096 --max_seq_len 5120 \
15 --use_paged_context_fmha enable1from tensorrt_llm.runtime import ModelRunner
2from transformers import AutoTokenizer
3
4tok = AutoTokenizer.from_pretrained("tss-deposium/gemma-4-E4B-text-only-nvfp4")
5runner = ModelRunner.from_dir("./engine")
6
7prompt = tok.apply_chat_template(
8 [{"role": "user", "content": "Quelle est la capitale de la France ?"}],
9 tokenize=False, add_generation_prompt=True,
10)
11ids = tok(prompt, return_tensors="pt").input_ids.cuda()
12out = runner.generate(ids, max_new_tokens=64)
13print(tok.decode(out[0][0]))tss-deposium/gemma-4-E4B-text-only-onnx-int4| This repo (NVFP4) | Sibling INT4 ONNX | |
|---|---|---|
| Hardware | Blackwell only | Any GPU + CPU fallback |
| Stack | TensorRT-LLM | ONNX Runtime |
| Vitesse | 1.5-3× INT4 ONNX on Blackwell | baseline |
| Portabilité | Self-hosted RTX 50xx only | Linux/Docker/Railway/cross-OS |
| Quality | ~97-99% MMLU | ~95-97% MMLU |
google/gemma-4-E4B-it (multimodal — text decoder loaded directly via AutoModelForCausalLM)nvidia-modelopt NVFP4_DEFAULT_CFGdocs/gemma4_e4b_nvfp4_modelopt_export.ipynb