@coreviz/sdk 1.0.7 → 1.0.9

package/dist/coreviz.js

@@ -176,15 +176,17 @@ class CoreViz {
             // Dynamic import to avoid loading transformers if not used
             const { AutoTokenizer, AutoProcessor, CLIPTextModelWithProjection, CLIPVisionModelWithProjection, RawImage } = await Promise.resolve().then(() => __importStar(require('@huggingface/transformers')));
             const MODEL_ID = 'Xenova/clip-vit-large-patch14';
-            console.log(`Loading local model ${MODEL_ID}...`);
             const start = Date.now();
             // Load tokenizer and processor
             const tokenizer = await AutoTokenizer.from_pretrained(MODEL_ID);
             const processor = await AutoProcessor.from_pretrained(MODEL_ID);
             // Load models with device preference
-            const text_model = await CLIPTextModelWithProjection.from_pretrained(MODEL_ID);
-            const vision_model = await CLIPVisionModelWithProjection.from_pretrained(MODEL_ID);
-            console.log(`Model loaded in ${Date.now() - start}ms`);
+            const text_model = await CLIPTextModelWithProjection.from_pretrained(MODEL_ID, {
+                dtype: 'q4',
+            });
+            const vision_model = await CLIPVisionModelWithProjection.from_pretrained(MODEL_ID, {
+                dtype: 'q4',
+            });
             // Check if input is likely an image
             const isImage = options?.type === 'image' ||
                 input.startsWith('data:image') ||

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@coreviz/sdk",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "CoreViz SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",