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 FaceFiles (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