Add new smaller models

utils.js

@@ -94,14 +94,72 @@ const TASKS = {
 };
 
 export const MODELS = {
-  coedit_large_quantized_4k: {
-    size: "441 MB",
+  coedit_small_quantized_4_0: {
+    size: "43.4 MB",
     base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
-    model: "model-q4k.gguf",
+    model: "model-small-q4_0.gguf",
+    tokenizer: "tokenizer.json",
+    config: "config-small.json",
+    tasks: TASKS,
+  },
+  coedit_small_quantized_4k: {
+    size: "59.6 MB",
+    base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
+    model: "model-small-q4k.gguf",
+    tokenizer: "tokenizer.json",
+    config: "config-small.json",
+    tasks: TASKS,
+  },
+  coedit_small_quantized_6k: {
+    size: "78.2 MB",
+    base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
+    model: "model-small.gguf",
+    tokenizer: "tokenizer.json",
+    config: "config-small.json",
+    tasks: TASKS,
+  },
+  coedit_small_fp32: {
+    size: "308 MB",
+    base_url: "https://huggingface.co/jbochi/coedit-base/resolve/main/",
+    model: "model.safetensors",
     tokenizer: "tokenizer.json",
     config: "config.json",
     tasks: TASKS,
   },
+
+  coedit_base_quantized_4_0: {
+    size: "139 MB",
+    base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
+    model: "model-base-q4_0.gguf",
+    tokenizer: "tokenizer.json",
+    config: "config-base.json",
+    tasks: TASKS,
+  },
+  coedit_base_quantized_4k: {
+    size: "139 MB",
+    base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
+    model: "model-base-q4k.gguf",
+    tokenizer: "tokenizer.json",
+    config: "config-base.json",
+    tasks: TASKS,
+  },
+  coedit_base_quantized_6k: {
+    size: "203 MB",
+    base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
+    model: "model-base.gguf",
+    tokenizer: "tokenizer.json",
+    config: "config-base.json",
+    tasks: TASKS,
+  },
+  coedit_base_fp32: {
+    size: "990 MB",
+    base_url: "https://huggingface.co/jbochi/coedit-base/resolve/main/",
+    model: "model.safetensors",
+    tokenizer: "tokenizer.json",
+    config: "config.json",
+    tasks: TASKS,
+  },
+
   coedit_large_quantized_4_0: {
     size: "441 MB",
     base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
@@ -110,22 +168,23 @@ export const MODELS = {
     config: "config.json",
     tasks: TASKS,
   },
-  coedit_large_quantized_6k: {
-    size: "643 MB",
+  coedit_large_quantized_4k: {
+    size: "441 MB",
     base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
-    model: "model.gguf",
+    model: "model-q4k.gguf",
     tokenizer: "tokenizer.json",
     config: "config.json",
     tasks: TASKS,
-  },
-  coedit_xl_quantized_4k: {
-    size: "1.6 GB",
+  },  
+  coedit_large_quantized_6k: {
+    size: "643 MB",
     base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
-    model: "model-xl-q4k.gguf",
+    model: "model.gguf",
     tokenizer: "tokenizer.json",
-    config: "config-xl.json",
+    config: "config.json",
     tasks: TASKS,
   },
+  
   coedit_xl_quantized_4_0: {
     size: "1.6 GB",
     base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
@@ -134,22 +193,15 @@ export const MODELS = {
     config: "config.json",
     tasks: TASKS,
   },
-  coedit_xl_quantized_6k: {
-    size: "2.34 GB",
+  coedit_xl_quantized_4k: {
+    size: "1.6 GB",
     base_url: "https://huggingface.co/jbochi/candle-coedit-quantized/resolve/main/",
-    model: "model-xl.gguf",
+    model: "model-xl-q4k.gguf",
     tokenizer: "tokenizer.json",
     config: "config-xl.json",
     tasks: TASKS,
   },
-  coedit_large: {
-    size: "3.13 GB",
-    base_url: "https://huggingface.co/grammarly/coedit-large/resolve/main/",
-    model: "model.safetensors",
-    tokenizer: "tokenizer.json",
-    config: "config.json",
-    tasks: TASKS,
-  },
+  
 };
 
 export function getModelInfo(id, taskID) {