ONNX Model Zoo

Felladrin/onnx-TinyMistral-248M-SFT-v4

By Felladrin · ↓ 2 · ♥ 0 · Updated 2024-01-07

What's in this repo

This page lists the files in Felladrin/onnx-TinyMistral-248M-SFT-v4, including decoder_model_merged.onnx, decoder_model_merged_quantized.onnx.

  • 2 ONNX models (.onnx files)
  • 10 files, including 6 .json, 2 .onnx, 1 .md
  • Last updated 2024-01-07

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

INT8 ONNX version of Felladrin/TinyMistral-248M-SFT-v4 to use with Transformers.js.

Usage

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

const prompt = `<|im_start|>user
Invited some friends to come home today. Give me some ideas for games to play with them!<|im_end|>
<|im_start|>assistant`;
const generator = await pipeline('text-generation', 'Felladrin/onnx-TinyMistral-248M-SFT-v4');
const output = await generator(prompt, { add_special_tokens: true, max_new_tokens: 128, do_sample: true, penalty_alpha: 0.6, top_k: 4, repetition_penalty: 1.07 });
console.log(output[0].generated_text);
// Sure! Here are a few suggestions for activities that could be added to a game night: 
// 1. Play board games and puzzle games 
// 2. Watch the movie trailer or gameplay videos 
// 3. Play a board game tournament or tournament with friends 
// 4. Watch movies on video or TV shows like Netflix, Hulu, and IFC.<|im_end|>

Model info

text-generation transformers.js
transformers.js onnx mistral text-generation conversational base_model:Felladrin/TinyMistral-248M-Chat-v4 base_model:quantized:Felladrin/TinyMistral-248M-Chat-v4 license:apache-2.0 region:us
↓ 2 ♥ 0

Created: 2023-12-11

Updated: 2024-01-07

View on Hugging Face

Files (10)

.gitattributes
README.md
config.json
generation_config.json
onnx/decoder_model_merged.onnx ONNX
quantize_config.json
special_tokens_map.json
tokenizer.json
tokenizer_config.json

Related models