Skip to content

Commit

Permalink
cargo fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
jorgeantonio21 committed Apr 2, 2024
1 parent 9696f22 commit 32975c3
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
2 changes: 1 addition & 1 deletion atoma-inference/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ async fn main() -> Result<(), ModelServiceError> {
req_sender
.send(TextRequest {
request_id: 0,
prompt: "Who was the first american president ?".to_string(),
prompt: "Leon, the professional is a movie".to_string(),
model: "state-spaces/mamba-130m".to_string(),
max_tokens: 512,
temperature: Some(0.0),
Expand Down
6 changes: 2 additions & 4 deletions atoma-inference/src/service.rs
Original file line number Diff line number Diff line change
Expand Up @@ -188,8 +188,7 @@ mod tests {
impl Request for () {
type ModelInput = ();

fn into_model_input(self) -> Self::ModelInput {
}
fn into_model_input(self) -> Self::ModelInput {}

fn is_node_authorized(&self, _: &PublicKey) -> bool {
true
Expand All @@ -207,8 +206,7 @@ mod tests {
impl Response for () {
type ModelOutput = ();

fn from_model_output(_: Self::ModelOutput) -> Self {
}
fn from_model_output(_: Self::ModelOutput) -> Self {}
}

#[derive(Clone)]
Expand Down

0 comments on commit 32975c3

Please sign in to comment.