Emalawi19 commited on
Commit
8fcc986
·
verified ·
1 Parent(s): 5171cc2

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +4 -2
server.js CHANGED
@@ -3,13 +3,15 @@ import http from 'http';
3
 
4
  // Configuration
5
  const PORT = 7860;
6
- const MODEL_NAME = 'Xenova/starcoderbase-1b'; // ← 1B param, public, strong code generation
7
  let generator;
8
 
9
  // Initialize the model on startup
10
  async function loadModel() {
11
  console.log("Loading coding model...");
12
- generator = await pipeline('text-generation', MODEL_NAME);
 
 
13
  console.log("Model loaded successfully!");
14
  }
15
 
 
3
 
4
  // Configuration
5
  const PORT = 7860;
6
+ const MODEL_NAME = 'Xenova/codegen-350M-multi'; // ← Multi-language: JS, Python, C++, etc.
7
  let generator;
8
 
9
  // Initialize the model on startup
10
  async function loadModel() {
11
  console.log("Loading coding model...");
12
+ generator = await pipeline('text-generation', MODEL_NAME, {
13
+ model_file_name: 'decoder_model_merged' // ← Required for codegen models in Transformers.js v3
14
+ });
15
  console.log("Model loaded successfully!");
16
  }
17