pub struct Training {
    pub parent: Config,
}

Fields§

§parent: Config

Implementations§

source§

impl Training

Training struct contains all the functionality for interacting with the training endpoints of the Replicate API.

source

pub fn new(rep: Config) -> Self

source

pub fn create( &self, model_owner: String, model_name: String, version_id: String, options: TrainingOptions ) -> Result<CreateTraining, Box<dyn Error>>

Create a new training.

Arguments
  • model_owner - The name of the user or organization that owns the model.
  • model_name - The name of the model.
  • version_id - The ID of the version.
  • options - The options for creating a training.
    • destination - A string representing the desired model to push to in the format {destination_model_owner}/{destination_model_name}. This should be an existing model owned by the user or organization making the API request. If the destination is invalid, the server returns an appropriate 4XX response.
    • input - An object containing inputs to the Cog model’s train() function.
    • webhook - An HTTPS URL for receiving a webhook when the training completes. The webhook will be a POST request where the request body is the same as the response body of the get training operation. If there are network problems, we will retry the webhook a few times, so make sure it can be safely called more than once.
  • _webhook_events_filter - TO only send specifc events to the webhook, use this field. If not specified, all events will be sent. The following events are supported:
Example
use replicate_rust::{Replicate, config::Config, training::TrainingOptions};
 
let config = Config::default();
let replicate = Replicate::new(config);
 
let mut input = HashMap::new();
input.insert(String::from("training_data"), String::from("https://example.com/70k_samples.jsonl"));
 
let result = replicate.trainings.create(
   String::from("owner"),
   String::from("model"),
  String::from("632231d0d49d34d5c4633bd838aee3d81d936e59a886fbf28524702003b4c532"),
 TrainingOptions {
    destination: String::from({new_owner}/{new_name}),
    input,
    webhook: String::from("https://example.com/my-webhook"),
    _webhook_events_filter: None,
},
);
source

pub fn get(&self, training_id: String) -> Result<GetTraining, Box<dyn Error>>

Get the details of a training.

Arguments
  • training_id - The ID of the training you want to get.
Example
use replicate_rust::{Replicate, config::Config};
 
let config = Config::default();
let replicate = Replicate::new(config);
 
match replicate.trainings.get(String::from("zz4ibbonubfz7carwiefibzgga")) {
  Ok(result) => println!("Success : {:?}", result),
  Err(e) => println!("Error : {}", e),
};
source

pub fn list(&self) -> Result<ListTraining, Box<dyn Error>>

Get a paginated list of trainings that you’ve created with your account. Returns 100 records per page.

Example
use replicate_rust::{Replicate, config::Config};
 
let config = Config::default();
let replicate = Replicate::new(config);
 
match replicate.trainings.list() {
    Ok(result) => println!("Success : {:?}", result),
    Err(e) => println!("Error : {}", e),
};
source

pub fn cancel(&self, training_id: String) -> Result<GetTraining, Box<dyn Error>>

Cancel a training.

Arguments
  • training_id - The ID of the training you want to cancel.
Example
use replicate_rust::{Replicate, config::Config};
 
let config = Config::default();
let replicate = Replicate::new(config);
 
match replicate.trainings.cancel(String::from("zz4ibbonubfz7carwiefibzgga")) {
    Ok(result) => println!("Success : {:?}", result),
   Err(e) => println!("Error : {}", e),
};

Auto Trait Implementations§

Blanket Implementations§

source§

impl<T> Any for Twhere T: 'static + ?Sized,

source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
source§

impl<T> Borrow<T> for Twhere T: ?Sized,

source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
source§

impl<T> BorrowMut<T> for Twhere T: ?Sized,

source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
source§

impl<T> From<T> for T

source§

fn from(t: T) -> T

Returns the argument unchanged.

source§

impl<T> Instrument for T

source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
source§

impl<T, U> Into<U> for Twhere U: From<T>,

source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

source§

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

§

type Error = Infallible

The type returned in the event of a conversion error.
source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
source§

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
source§

impl<T> WithSubscriber for T

source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more