fmt

coasys · Dec 17, 2024 · 8c7fc27 · 8c7fc27
1 parent bf7ee15
commit 8c7fc27
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/rust-executor/src/ai_service/mod.rs b/rust-executor/src/ai_service/mod.rs
@@ -302,10 +302,16 @@ impl AIService {
             "llama_34b_code" => Llama::builder().with_source(LlamaSource::llama_34b_code()),
             "llama_70b" => Llama::builder().with_source(LlamaSource::llama_70b()),
             "mistral_7b" => Llama::builder().with_source(LlamaSource::mistral_7b()),
-            "mistral_7b_instruct" => Llama::builder().with_source(LlamaSource::mistral_7b_instruct()),
-            "mistral_7b_instruct_2" => Llama::builder().with_source(LlamaSource::mistral_7b_instruct_2()),
+            "mistral_7b_instruct" => {
+                Llama::builder().with_source(LlamaSource::mistral_7b_instruct())
+            }
+            "mistral_7b_instruct_2" => {
+                Llama::builder().with_source(LlamaSource::mistral_7b_instruct_2())
+            }
             "solar_10_7b" => Llama::builder().with_source(LlamaSource::solar_10_7b()),
-            "solar_10_7b_instruct" => Llama::builder().with_source(LlamaSource::solar_10_7b_instruct()),
+            "solar_10_7b_instruct" => {
+                Llama::builder().with_source(LlamaSource::solar_10_7b_instruct())
+            }
 
             // Handle unknown models
             _ => {