pub struct JambaModelV2 { /* private fields */ }Expand description
Main Jamba model implementation
Trait Implementations§
Source§impl Model for JambaModelV2
impl Model for JambaModelV2
Source§type Config = JambaConfig
type Config = JambaConfig
Model configuration type
Source§fn new(config: JambaConfig) -> Result<Self>
fn new(config: JambaConfig) -> Result<Self>
Create new model instance
Source§fn from_weights(config: JambaConfig, weights: ModelWeights) -> Result<Self>
fn from_weights(config: JambaConfig, weights: ModelWeights) -> Result<Self>
Load model with weights
Source§fn forward(&self, inputs: &ModelInputs) -> Result<ModelOutputs>
fn forward(&self, inputs: &ModelInputs) -> Result<ModelOutputs>
Forward pass - core inference method
Source§fn generate(&self, prompt: &str, config: &GenerationConfig) -> Result<String>
fn generate(&self, prompt: &str, config: &GenerationConfig) -> Result<String>
Generate text (high-level interface)
Source§fn memory_requirements(&self) -> MemoryRequirements
fn memory_requirements(&self) -> MemoryRequirements
Get model memory requirements
Auto Trait Implementations§
impl !RefUnwindSafe for JambaModelV2
impl !UnwindSafe for JambaModelV2
impl Freeze for JambaModelV2
impl Send for JambaModelV2
impl Sync for JambaModelV2
impl Unpin for JambaModelV2
impl UnsafeUnpin for JambaModelV2
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more