content: [package]
  authors = ["Aasheesh Singh aasheeshdtu@gmail.com"]
  license = "MIT OR Apache-2.0"
  name = "bert-burn"
  version = "0.2.0"
  edition = "2021"
  
  [features]
  default = []
  f16 = []
  ndarray = ["burn/ndarray"]
  tch-cpu = ["burn/tch"]
  tch-gpu = ["burn/tch"]
  wgpu = ["burn/wgpu"]
  cuda = ["burn/cuda"]
  fusion = ["burn/fusion"]
  # To be replaced by burn-safetensors once supported: https://github.com/tracel-ai/burn/issues/626
  safetensors = ["candle-core/default"]
  
  
  [dependencies]
  # Burn
  burn = { version = "0.18.0", default-features = false, features = ["dataset", "std"] }
  cubecl-runtime = { version = "0.3.0", features = ["channel-mpsc"] } # missing feature flag when burn default-features are off
  candle-core = { version = "0.8.4" }
  # Tokenizer
  tokenizers = { version = "0.15.0", default-features = false, features = [
    "onig",
    "http",
  ] }
  derive-new = "0.7.0"
  hf-hub = { version = "0.3.2", features = ["tokio"] }
  
  # Utils
  serde = { version = "1.0.196", features = ["std", "derive"] }
  libm = "0.2.8"
  serde_json = "1.0.113"
  tokio = "1.35.1"
  