Skip to main content

Crate wax_core

Crate wax_core 

Source

Re-exports§

pub use device::DTypeChoice;
pub use device::DeviceChoice;
pub use engine::Engine;
pub use engine::EngineConfig;
pub use engine::GenerateRequest;
pub use engine::StreamSink;
pub use error::Result;
pub use error::WaxError;
pub use loader::resolve_model_source;
pub use loader::resolve_safetensors_files;
pub use loader::ModelConfig;
pub use loader::ModelSource;
pub use sampler::SamplingConfig;

Modules§

device
engine
error
loader
sampler
stats
token_stream

Enums§

DType
The different types of elements allowed in tensors.
Device
Cpu, Cuda, or Metal