Registry / other / candle-transformers

candle-transformers

library0.10.2rscratesunverified

Transformer model implementations for the Candle ML framework.

# Cargo.toml
[dependencies]
candle-transformers = "0.10.2"

github ↗homepage ↗docs ↗

INSTALL

IMPORT

SIG · CANDLE-TRANSFORMER

candle-transformers

otherrustv0.10.2

harness data pending

Install & Compatibility

Where this runs

No compatibility data collected yet for this library.

Code

Verified usage

Verified import paths — ran on the pinned version, not inferred.

BertModel

✓ use candle_transformers::models::bert::BertModel;

Creates a BERT transformer model on CPU.

use candle_transformers::models::bert::BertModel; use candle_core::Device; fn main() -> Result<(), Box<dyn std::error::Error>> { let device = Device::Cpu; let model = BertModel::new(&Default::default(), 768, 12, 12, 30522, 0.1)?; println!("BERT model created on CPU"); Ok(()) }

Debug

Known issues

gotchaRequires the `candle-core` and `candle-nn` crates as dependencies.

fix

Add `candle-core` and `candle-nn` to Cargo.toml.

affects: >=0.1.0

Upgrade

Version history

0.10.2latest on crates.io

Audit

Dependencies

No dependency data recorded yet.

Agent activity

7 hits · last 30 days

node

Amazon

Resources

githubhttps://github.com/huggingface/candle ↗

homepagehttps://github.com/huggingface/candle ↗

docshttps://docs.rs/candle-transformers/ ↗