Skip to content

Commit

Permalink
Add V7 to ModelVersion.
Browse files Browse the repository at this point in the history
  • Loading branch information
cryscan committed Nov 30, 2024
1 parent 6660386 commit af8d9c3
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 0 deletions.
1 change: 1 addition & 0 deletions examples/batch.rs
Original file line number Diff line number Diff line change
Expand Up @@ -207,6 +207,7 @@ async fn main() -> Result<()> {
let bundle = v6::Bundle::<f16>::new(model, cli.batch);
TokioRuntime::new(bundle).await
}
ModelVersion::V7 => todo!(),
};

#[cfg(not(debug_assertions))]
Expand Down
1 change: 1 addition & 0 deletions examples/chat.rs
Original file line number Diff line number Diff line change
Expand Up @@ -274,6 +274,7 @@ async fn main() -> Result<()> {
let state = builder.state();
(TokioRuntime::new(builder).await, Box::new(state))
}
ModelVersion::V7 => todo!(),
};

// run initial prompt
Expand Down
1 change: 1 addition & 0 deletions examples/gen.rs
Original file line number Diff line number Diff line change
Expand Up @@ -187,6 +187,7 @@ async fn main() -> Result<()> {
let bundle = v6::Bundle::<f16>::new(model, 1);
TokioRuntime::new(bundle).await
}
ModelVersion::V7 => todo!(),
};

// const PROMPT: &str = "User: Hi!\n\nAssistant: Hello! I'm your AI assistant. I'm here to help you with various tasks, such as answering questions, brainstorming ideas, drafting emails, writing code, providing advice, and much more.\n\nUser: Hi!\n\nAssistant:";
Expand Down
1 change: 1 addition & 0 deletions examples/serialization.rs
Original file line number Diff line number Diff line change
Expand Up @@ -203,6 +203,7 @@ async fn main() -> Result<()> {
let bundle = v6::Bundle::<f16>::new(model, 1);
TokioRuntime::new(bundle).await
}
ModelVersion::V7 => todo!(),
};

const PROMPT: &str = include_str!("prompt.md");
Expand Down
1 change: 1 addition & 0 deletions src/runtime/model.rs
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ pub enum ModelVersion {
V4,
V5,
V6,
V7,
}

#[wasm_bindgen]
Expand Down

0 comments on commit af8d9c3

Please sign in to comment.