forked from tracel-ai/burn
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Cargo.toml
41 lines (35 loc) · 1.05 KB
/
Cargo.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
[package]
authors = ["Dilshod Tadjibaev (@antimora)"]
edition.workspace = true
license = "MIT OR Apache-2.0"
name = "image-classification-web"
publish = false
version.workspace = true
[lib]
crate-type = ["cdylib"]
[features]
default = []
half_precision = []
[dependencies]
burn = { path = "../../crates/burn", version = "0.14.0", default-features = false, features = [
"ndarray",
] }
cubecl = { workspace = true, features = ["wgpu"] }
burn-wgpu = { path = "../../crates/burn-wgpu", version = "0.14.0", default-features = false, features = [
"autotune",
] }
burn-candle = { path = "../../crates/burn-candle", version = "0.14.0", default-features = false }
log = { workspace = true }
serde = { workspace = true }
serde_json = { workspace = true }
console_error_panic_hook = { workspace = true }
# Wasm dependencies
wasm-bindgen = "0.2"
wasm-bindgen-futures = "0.4"
serde-wasm-bindgen = "0.6"
wasm-logger = "0.2"
wasm-timer = "0.2"
js-sys = "0.3"
[build-dependencies]
# Used to generate code from ONNX model
burn-import = { path = "../../crates/burn-import" }