Skip to content

Commit a3c600e

Browse files
committed
add qwen models
1 parent c464cbf commit a3c600e

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

executor/src/models.rs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,12 @@ pub enum Model {
2626
/// [Google's Gemma3 27b](https://ollama.com/library/gemma3:27b)
2727
#[serde(rename = "gemma3:27b")]
2828
Gemma3_27b,
29+
/// [Alibaba's Qwen3 32b](https://ollama.com/library/qwen3:32b)
30+
#[serde(rename = "qwen3:32b")]
31+
Qwen3_32b,
32+
/// [Alibaba's Qwen3 8b](https://ollama.com/library/qwen3:8b)
33+
#[serde(rename = "qwen3:8b")]
34+
Qwen3_8b,
2935
// // OpenAI models
3036
// /// [OpenAI's GPT-4o](https://platform.openai.com/docs/models#gpt-4o)
3137
// #[serde(rename = "gpt-4o")]
@@ -177,13 +183,15 @@ impl From<&Model> for ModelProvider {
177183
fn from(model: &Model) -> Self {
178184
match model {
179185
// ollama
186+
Model::Gemma3_4b => ModelProvider::Ollama,
180187
Model::Gemma3_12b => ModelProvider::Ollama,
181188
Model::Gemma3_27b => ModelProvider::Ollama,
182-
Model::Gemma3_4b => ModelProvider::Ollama,
183189
Model::Llama3_1_8bInstructQ4Km => ModelProvider::Ollama,
184190
Model::Llama3_2_1bInstructQ4Km => ModelProvider::Ollama,
185191
Model::Llama3_3_70bInstructQ4Km => ModelProvider::Ollama,
186192
Model::MistralNemo12b => ModelProvider::Ollama,
193+
Model::Qwen3_8b => ModelProvider::Ollama,
194+
Model::Qwen3_32b => ModelProvider::Ollama,
187195
// // openai
188196
// Model::GPT4o => ModelProvider::OpenAI,
189197
// Model::GPT4oMini => ModelProvider::OpenAI,

0 commit comments

Comments
 (0)