ONNX Model Zoo

Xenova/stablelm-2-zephyr-1_6b

By Xenova · ↓ 8 · ♥ 1 · Updated 2024-10-08

What's in this repo

This page lists the files in Xenova/stablelm-2-zephyr-1_6b, including decoder_model_merged.onnx, decoder_model_merged_quantized.onnx.

  • 2 ONNX models (.onnx files)
  • 14 files, including 7 .json, 2 .onnx, 1 .md, 1 .py, 1 .txt, 1 .onnx_data
  • Last updated 2024-10-08

ONNX files

File Size Download
onnx/decoder_model_merged.onnx — Download
onnx/decoder_model_merged_quantized.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

https://huggingface.co/stabilityai/stablelm-2-zephyr-1_6b with ONNX weights to be compatible with Transformers.js.

Usage (Transformers.js)

If you haven't already, you can install the Transformers.js JavaScript library from NPM using:

npm i @xenova/transformers

Example: Text generation with Xenova/stablelm-2-zephyr-1_6b.

import { pipeline } from '@xenova/transformers';

// Create text generation pipeline
const generator = await pipeline('text-generation', 'Xenova/stablelm-2-zephyr-1_6b');

// Define the prompt and list of messages
const prompt = "Tell me a funny joke."
const messages = [
    { "role": "system", "content": "You are a helpful assistant." },
    { "role": "user", "content": prompt },
]

// Apply chat template
const inputs = generator.tokenizer.apply_chat_template(messages, {
    tokenize: false,
    add_generation_prompt: true,
});

// Generate text
const output = await generator(inputs, { max_new_tokens: 20 });
console.log(output[0].generated_text);
// "<|system|>\nYou are a helpful assistant.\n<|user|>\nTell me a funny joke.\n<|assistant|>\nHere's a joke for you:\n\nWhy don't scientists trust atoms?\n\nBecause they make up everything!"

Note: Having a separate repo for ONNX weights is intended to be a temporary solution until WebML gains more traction. If you would like to make your models web-ready, we recommend converting to ONNX using 🤗 Optimum and structuring your repo like this one (with ONNX weights located in a subfolder named onnx).

Model info

text-generation transformers.js
transformers.js onnx stablelm text-generation conversational base_model:stabilityai/stablelm-2-zephyr-1_6b base_model:quantized:stabilityai/stablelm-2-zephyr-1_6b region:us
↓ 8 ♥ 1

Created: 2024-02-02

Updated: 2024-10-08

View on Hugging Face

Files (14)

.gitattributes
README.md
config.json
configuration_stablelm_epoch.py
generation_config.json
merges.txt
onnx/decoder_model_merged.onnx ONNX
onnx/decoder_model_merged.onnx_data
quantize_config.json
special_tokens_map.json
tokenizer.json
tokenizer_config.json
vocab.json

Related models