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
..
src bert attention mask (#1934) 2024-08-01 08:26:19 +02:00
Cargo.toml Update gloo requirement from 0.8 to 0.11 (#1558) 2024-01-10 16:27:20 +01:00
README.md BERT Wasm (#902) 2023-09-19 21:31:37 +01:00
bertWorker.js BERT Wasm (#902) 2023-09-19 21:31:37 +01:00
build-lib.sh BERT Wasm (#902) 2023-09-19 21:31:37 +01:00
lib-example.html BERT Wasm (#902) 2023-09-19 21:31:37 +01:00
utils.js BERT Wasm (#902) 2023-09-19 21:31:37 +01:00

README.md

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.