mobileclip-webnn-gpu

Running

Xenova HF Staff commited on 12 days ago

Commit

cab5a6c

verified ·

1 Parent(s): c6ea63d

Update main.js

Files changed (1) hide show

main.js CHANGED Viewed

@@ -31,7 +31,7 @@ try {
   // Load processor and vision model
   processor = await AutoProcessor.from_pretrained(model_id);
   vision_model = await CLIPVisionModelWithProjection.from_pretrained(model_id, {
-    device: "webnn",
     dtype: "fp32",
   });
 } catch (err) {

   // Load processor and vision model
   processor = await AutoProcessor.from_pretrained(model_id);
   vision_model = await CLIPVisionModelWithProjection.from_pretrained(model_id, {
+    device: "webnn-gpu",
     dtype: "fp32",
   });
 } catch (err) {