2023-04-15 22:44:50 +08:00
|
|
|
[package]
|
|
|
|
authors = [
|
2024-01-25 23:20:09 +08:00
|
|
|
"Dilshod Tadjibaev (@antimora)",
|
|
|
|
"Nathaniel Simard (@nathanielsimard)",
|
2023-04-15 22:44:50 +08:00
|
|
|
]
|
2023-06-15 21:20:53 +08:00
|
|
|
description = "Library for importing datamodels into the Burn framework"
|
2023-12-12 22:39:07 +08:00
|
|
|
edition.workspace = true
|
|
|
|
license.workspace = true
|
2023-04-15 22:44:50 +08:00
|
|
|
name = "burn-import"
|
2023-12-12 22:39:07 +08:00
|
|
|
readme.workspace = true
|
2024-05-10 03:40:05 +08:00
|
|
|
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-import"
|
2023-12-12 22:39:07 +08:00
|
|
|
version.workspace = true
|
2023-04-15 22:44:50 +08:00
|
|
|
|
2024-01-25 23:20:09 +08:00
|
|
|
default-run = "onnx2burn"
|
|
|
|
|
2023-04-15 22:44:50 +08:00
|
|
|
[features]
|
2024-01-25 23:20:09 +08:00
|
|
|
default = ["onnx", "pytorch"]
|
2023-04-15 22:44:50 +08:00
|
|
|
onnx = []
|
2024-02-20 23:52:36 +08:00
|
|
|
pytorch = ["burn/record-item-custom-serde", "thiserror", "zip"]
|
2023-04-15 22:44:50 +08:00
|
|
|
|
|
|
|
[dependencies]
|
2024-08-28 02:12:54 +08:00
|
|
|
burn = { path = "../burn", version = "0.15.0", features = ["ndarray"] }
|
|
|
|
onnx-ir = { path = "../onnx-ir", version = "0.15.0" }
|
2024-01-25 23:20:09 +08:00
|
|
|
candle-core = { workspace = true }
|
2023-12-02 03:33:28 +08:00
|
|
|
derive-new = { workspace = true }
|
|
|
|
half = { workspace = true }
|
|
|
|
log = { workspace = true }
|
|
|
|
proc-macro2 = { workspace = true }
|
|
|
|
quote = { workspace = true }
|
2024-01-25 23:20:09 +08:00
|
|
|
regex = { workspace = true }
|
|
|
|
rust-format = { workspace = true, features = ["token_stream", "post_process"] }
|
|
|
|
serde = { workspace = true, features = ["derive"] }
|
2023-12-02 03:33:28 +08:00
|
|
|
serde_json = { workspace = true, features = ["std"] }
|
|
|
|
syn = { workspace = true, features = ["parsing"] }
|
2024-01-25 23:20:09 +08:00
|
|
|
thiserror = { workspace = true, optional = true }
|
2023-12-12 22:39:07 +08:00
|
|
|
tracing-core = { workspace = true }
|
2024-01-25 23:20:09 +08:00
|
|
|
tracing-subscriber = { workspace = true }
|
2024-02-20 23:52:36 +08:00
|
|
|
zip = { workspace = true, optional = true }
|
2023-04-15 22:44:50 +08:00
|
|
|
|
|
|
|
[dev-dependencies]
|
2023-12-02 03:33:28 +08:00
|
|
|
pretty_assertions = { workspace = true }
|
2024-01-25 23:20:09 +08:00
|
|
|
rstest = { workspace = true }
|