Files
candle/candle-wasm-examples/bert
Zheng Li 4a52aeb437 bert attention mask (#1934)
* bert attention mask

* Allow for using None as a mask.

* Revert part of the changes so that the proper default mask applies.

* Cosmetic change.

* Another cosmetic tweak.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>
2024-08-01 08:26:19 +02:00
..
2024-08-01 08:26:19 +02:00
2023-09-19 21:31:37 +01:00
2023-09-19 21:31:37 +01:00
2023-09-19 21:31:37 +01:00
2023-09-19 21:31:37 +01:00
2023-09-19 21:31:37 +01:00

Running BERT with Candle and WASM

Here, we provide two examples of how to run Bert using a Candle-compiled WASM binary and runtime.

Vanilla JS and WebWorkers

To build and test the UI made in Vanilla JS and WebWorkers, first we need to build the WASM library:

sh build-lib.sh

This will bundle the library under ./build and we can import it inside our WebWorker like a normal JS module:

import init, { Model } from "./build/m.js";

The full example can be found under ./lib-example.html. All needed assets are fetched from the web, so no need to download anything. Finally, you can preview the example by running a local HTTP server. For example:

python -m http.server

Then open http://localhost:8000/lib-example.html in your browser.