PeterPinetree commited on
Commit
0e53e24
·
verified ·
1 Parent(s): 54c6c11

Update index.html

Browse files
Files changed (1) hide show
  1. index.html +8 -6
index.html CHANGED
@@ -200,11 +200,11 @@
200
  // --- Model registry ---
201
  const MODELS = {
202
  qwen: {
203
- id: "qwen", // this is the folder name under assets/models/
204
- base: new URL("assets/models/qwen/", window.location.href).href,
205
- onnx_file: "onnx/model_q4f16.onnx",
206
- emb_coords: "assets/embeddings/qwen_pca_top5k_coords.json",
207
- emb_nbrs: "assets/embeddings/qwen_neighbors_top5k_k40.json",
208
  },
209
  distilgpt2: {
210
  repo: "Xenova/distilgpt2",
@@ -346,9 +346,11 @@
346
  try {
347
  if (key === 'qwen') {
348
  model = await AutoModelForCausalLM.from_pretrained(MODELS.qwen.id, {
349
- file_name: MODELS.qwen.onnx_file,
 
350
  progress_callback: onProgress,
351
  config: QWEN3_CONFIG_FIX,
 
352
  });
353
  } else {
354
  model = await AutoModelForCausalLM.from_pretrained(MODELS.distilgpt2.repo, {
 
200
  // --- Model registry ---
201
  const MODELS = {
202
  qwen: {
203
+ id: 'qwen',
204
+ subfolder: 'onnx',
205
+ model_file_name: 'model_q4f16',
206
+ emb_coords: 'assets/embeddings/qwen_pca_top5k_coords.json',
207
+ emb_nbrs: 'assets/embeddings/qwen_neighbors_top5k_k40.json',
208
  },
209
  distilgpt2: {
210
  repo: "Xenova/distilgpt2",
 
346
  try {
347
  if (key === 'qwen') {
348
  model = await AutoModelForCausalLM.from_pretrained(MODELS.qwen.id, {
349
+ subfolder: MODELS.qwen.subfolder,
350
+ model_file_name: MODELS.qwen.model_file_name,
351
  progress_callback: onProgress,
352
  config: QWEN3_CONFIG_FIX,
353
+ local_files_only: true, // optional but explicit
354
  });
355
  } else {
356
  model = await AutoModelForCausalLM.from_pretrained(MODELS.distilgpt2.repo, {