Views
No views yet
1import { pipeline } from '@xenova/transformers';
2
3const generator = await pipeline('text-generation', 'd1g1mortal/Dolphin3.0-Qwen2.5-1.5B-ONNX', {
4 device: 'webgpu',
5});
6
7const messages = [
8 { role: "system", content: "You are Dolphin, a helpful AI assistant." },
9 { role: "user", content: "Write a short creative poem about the ocean." },
10];
11
12const prompt = generator.tokenizer.apply_chat_template(messages, {
13 tokenize: false,
14 add_generation_prompt: true,
15});
16
17const output = await generator(prompt, {
18 max_new_tokens: 256,
19 temperature: 0.7,
20 do_sample: true,
21});
22
23console.log(output[0].generated_text);