Views
No views yet
npm i @huggingface/transformers1import { pipeline, TextStreamer } from "@huggingface/transformers";
2
3// Create a text generation pipeline
4const generator = await pipeline(
5 "text-generation",
6 "onnx-community/ettin-decoder-17m-ONNX",
7 { dtype: "fp32" },
8);
9
10// Generate a response
11const text = "Q: What is the capital of France?\nA:";
12const output = await generator(text, {
13 max_new_tokens: 128,
14 streamer: new TextStreamer(generator.tokenizer, { skip_prompt: true, skip_special_tokens: true }),
15});
16console.log(output[0].generated_text);