Views
No views yet
npm i @xenova/transformersXenova/phi-1_5_dev.1import { pipeline } from '@xenova/transformers';
2
3// Create a text-generation pipeline
4const generator = await pipeline('text-generation', 'Xenova/phi-1_5_dev');
5
6// Construct prompt
7const prompt = `\`\`\`py
8import math
9def print_prime(n):
10 """
11 Print all primes between 1 and n
12 """`;
13
14// Generate text
15const result = await generator(prompt, {
16 max_new_tokens: 100,
17});
18console.log(result[0].generated_text);1import math
2def print_prime(n):
3 """
4 Print all primes between 1 and n
5 """
6 primes = []
7 for num in range(2, n+1):
8 is_prime = True
9 for i in range(2, int(math.sqrt(num))+1):
10 if num % i == 0:
11 is_prime = False
12 break
13 if is_prime:
14 primes.append(num)
15 print(primes)
16
17print_prime(20)[2, 3, 5, 7, 11, 13, 17, 19]onnx).