Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
joseph-predibase committed Jan 29, 2025
1 parent 75d727f commit 2cae328
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion launcher/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1002,7 +1002,10 @@ fn shard_manager(

// Parse the model_id to extract the model and revision (in model@revision format). If the model_id string doesn't
// encode a revision, return revision_fallback for revision (which may itself be None).
fn get_model_and_revision(model_id: &str, revision_fallback: Option<String>) -> (String, Option<String>) {
fn get_model_and_revision(
model_id: &str,
revision_fallback: Option<String>,
) -> (String, Option<String>) {
let mut parts = model_id.split('@');
let model_id = parts.next().unwrap().to_string();
let revision = parts.next().map(|s| s.to_string()).or(revision_fallback);
Expand Down

0 comments on commit 2cae328

Please sign in to comment.