pub struct FalconModel { /* private fields */ }Expand description
Falcon transformer model
Implementations§
Source§impl FalconModel
impl FalconModel
pub fn new(config: FalconConfig) -> Result<Self>
pub fn new_with_device(config: FalconConfig, device: Device) -> Result<Self>
pub fn device(&self) -> Device
pub fn config(&self) -> &FalconConfig
Trait Implementations§
Source§impl Layer for FalconModel
impl Layer for FalconModel
Source§impl Model for FalconModel
impl Model for FalconModel
type Config = FalconConfig
type Input = Tensor
type Output = Tensor
Source§fn forward(&self, input: Self::Input) -> Result<Self::Output>
fn forward(&self, input: Self::Input) -> Result<Self::Output>
Performs a forward pass through the model. Read more
Source§fn load_pretrained(&mut self, _reader: &mut dyn Read) -> Result<()>
fn load_pretrained(&mut self, _reader: &mut dyn Read) -> Result<()>
Loads pretrained weights into the model. Read more
Source§fn get_config(&self) -> &Self::Config
fn get_config(&self) -> &Self::Config
Returns a reference to the model’s configuration. Read more
Source§fn num_parameters(&self) -> usize
fn num_parameters(&self) -> usize
Returns the total number of parameters in the model. Read more
Auto Trait Implementations§
impl Freeze for FalconModel
impl RefUnwindSafe for FalconModel
impl Send for FalconModel
impl Sync for FalconModel
impl Unpin for FalconModel
impl UnsafeUnpin for FalconModel
impl UnwindSafe for FalconModel
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
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