tracel-ai / models

Models and examples built with Burn
Apache License 2.0
169 stars 23 forks source link

[ResNet] Chore: Use Candle's published crate version instead of git revision #19

Closed laggui closed 5 months ago

laggui commented 7 months ago

Currently, the Cargo.toml points to a git revision of Burn required to properly import PyTorch weights.

Once candle-core includes the patches in a release version and Burn includes that version in a release, we should update the versions.

[dependencies]
# Note: default-features = false is needed to disable std
burn = { version = "0.13.0", default-features = false }
burn-import = "0.13.0"
dirs = { version = "5.0.1", optional = true }
serde = { version = "1.0.192", default-features = false, features = [
    "derive",
    "alloc",
] } # alloc is for no_std, derive is needed

[dev-dependencies]
burn = {  version = "0.13.0", features = ["ndarray"] }
image = { version = "0.24.7", features = ["png", "jpeg"] }