Skip to content

Commit

Permalink
Merge pull request #27 from anilaltuner/main
Browse files Browse the repository at this point in the history
  • Loading branch information
andthattoo authored Dec 6, 2024
2 parents 5b26cb4 + e6a6fbb commit 2399976
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions src/program/models.rs
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,9 @@ pub enum Model {
/// [Meta's LLama3.2 Edge models](https://ollama.com/library/llama3.2/tags), 3B parameters
#[serde(rename = "llama3.2:3b")]
Llama3_2_3B,
/// [Meta's LLama3.3 Edge models](https://ollama.com/library/llama3.3/tags), 3B parameters
#[serde(rename = "llama3.3:70b")]
Llama3_3_70B,
/// [Meta's LLama3.2 Edge models](https://ollama.com/library/llama3.2/tags), 1B parameters, q4
#[serde(rename = "llama3.2:1b-text-q4_K_M")]
Llama3_2_1BTextQ4KM,
Expand Down Expand Up @@ -156,6 +159,9 @@ pub enum Model {
#[serde(rename = "meta-llama/llama-3.1-70b-instruct:free")]
ORLlama3_1_70BFree,

#[serde(rename = "meta-llama/llama-3.3-70b-instruct")]
ORLlama3_3_70B,

#[serde(rename = "anthropic/claude-3.5-sonnet:beta")]
OR3_5Sonnet,

Expand Down Expand Up @@ -272,6 +278,7 @@ impl From<Model> for ModelProvider {
Model::Llama3_2_1B => ModelProvider::Ollama,
Model::Llama3_2_3B => ModelProvider::Ollama,
Model::Llama3_2_1BTextQ4KM => ModelProvider::Ollama,
Model::Llama3_3_70B => ModelProvider::Ollama,
Model::Gemma2_9B => ModelProvider::Ollama,
Model::Gemma2_9BFp16 => ModelProvider::Ollama,
Model::Qwen2_5_7B => ModelProvider::Ollama,
Expand Down Expand Up @@ -303,6 +310,7 @@ impl From<Model> for ModelProvider {
Model::ORLlama3_1_70B => ModelProvider::OpenRouter,
Model::ORLlama3_1_405B => ModelProvider::OpenRouter,
Model::ORLlama3_1_70BFree => ModelProvider::OpenRouter,
Model::ORLlama3_3_70B => ModelProvider::OpenRouter,
Model::ORQwen2_5Coder32B => ModelProvider::OpenRouter,
Model::ORQwen2_5_7B => ModelProvider::OpenRouter,
Model::ORQwen2_5_72B => ModelProvider::OpenRouter,
Expand Down

0 comments on commit 2399976

Please sign in to comment.