Update index.html
Browse files- index.html +8 -6
index.html
CHANGED
|
@@ -200,11 +200,11 @@
|
|
| 200 |
// --- Model registry ---
|
| 201 |
const MODELS = {
|
| 202 |
qwen: {
|
| 203 |
-
id:
|
| 204 |
-
|
| 205 |
-
|
| 206 |
-
emb_coords:
|
| 207 |
-
emb_nbrs:
|
| 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 |
-
|
|
|
|
| 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, {
|